| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/inputs/ |
| H A D | trivial_test_graph_input_yielder.cc | 31 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef() function
103 r.graph = CreateGraphDef(num_stages_, width_, tensor_size_, in NextItem()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_activity_listener_test.cc | 78 GraphDef CreateGraphDef() { in CreateGraphDef() function
93 GraphDef graph_def = CreateGraphDef(); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_test.cc | 668 test::CreateGraphDef(num_nodes, num_edges_per_node); in TEST_F()
681 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_InEdgeIteration()
721 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_GraphCreation()
760 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_ToGraphDef()
799 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_RemoveNode()
|
| H A D | benchmark_testlib.h | 72 GraphDef CreateGraphDef(int num_nodes, int num_edges_per_node) { in CreateGraphDef() function
|
| H A D | algorithm_test.cc | 210 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_PruneForReverseReachability()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session_test.cc | 44 void CreateGraphDef(GraphDef* graph_def, string node_names[3]) { in CreateGraphDef() function
91 CreateGraphDef(&graph, node_names); in TEST()
127 CreateGraphDef(&graph, node_names); in TEST()
172 CreateGraphDef(&graph, node_names); in TEST()
203 CreateGraphDef(&graph, node_names); in TEST()
235 CreateGraphDef(&graph, node_names); in TEST()
265 CreateGraphDef(&graph, node_names); in TEST()
292 CreateGraphDef(&graph, node_names); in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpcbench_test.cc | 124 GraphDef CreateGraphDef(int num_stages, int width, int tensor_size, in CreateGraphDef() function
176 GraphDef def = CreateGraphDef(num_stages, width, tensor_size, in BM_Helper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | grpc_session_debug_test.cc | 52 void CreateGraphDef(GraphDef* graph_def, string node_names[3]) { in CreateGraphDef() function
154 CreateGraphDef(&graph, node_names); in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | graph_view_test.cc | 300 test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_GraphViewConstruction()
341 test::CreateGraphDef(num_nodes, /*num_edges_per_node=*/16); in BM_GraphViewGetNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/ |
| H A D | graph_view_test.cc | 2398 GraphDef graph_def = test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_GraphViewTConstruction()
2418 GraphDef graph_def = test::CreateGraphDef(num_nodes, num_edges_per_node); in BM_MutableGraphViewClearAttrs()
2480 test::CreateGraphDef(num_nodes, /*num_edges_per_node=*/16); in BM_GraphViewTGetNode()
|