Searched refs:tensor_array_node (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | backports_test.cc | 118 NodeDef* tensor_array_node = graph_def.add_node(); in TEST() local
119 tensor_array_node->set_name("tensor_array_node"); in TEST()
120 tensor_array_node->set_op("TensorArrayV3"); in TEST()
121 tensor_array_node->add_input("size_node"); in TEST()
122 SetNodeAttr("dtype", DT_FLOAT, tensor_array_node); in TEST()
123 SetNodeAttr("element_shape", TensorShape({1, 2}), tensor_array_node); in TEST()
124 SetNodeAttr("dynamic_size", false, tensor_array_node); in TEST()
125 SetNodeAttr("clear_after_read", true, tensor_array_node); in TEST()
126 SetNodeAttr("tensor_array_name", "some_name", tensor_array_node); in TEST()
|