| 2722 | } |
| 2723 | |
| 2724 | FunctionDef NestedCPUResourceOutput() { |
| 2725 | return FDH::Create( |
| 2726 | // Name |
| 2727 | "NestedCPUResourceOutput", |
| 2728 | // Args |
| 2729 | {"x: float"}, |
| 2730 | // Return values |
| 2731 | {"ds: resource", "x_out: float"}, |
| 2732 | // Attr def |
| 2733 | {}, |
| 2734 | // Nodes |
| 2735 | { |
| 2736 | {{"y"}, |
| 2737 | "PartitionedCall", |
| 2738 | {"x"}, |
| 2739 | {{"Tin", DataTypeSlice{DT_FLOAT}}, |
| 2740 | {"Tout", DataTypeSlice{DT_RESOURCE, DT_FLOAT}}, |
| 2741 | {"f", FDH::FunctionRef("CPUResourceOutput", {})}}}, |
| 2742 | }, |
| 2743 | {{"ds", "y:output:0"}, {"x_out", "y:output:1"}}); |
| 2744 | } |
| 2745 | |
| 2746 | TEST_F(NestedPlacerTest, NestedDeepDeviceConstraintsPropagated) { |
| 2747 | /* |