Home
last modified time | relevance | path

Searched refs:subgraph_def (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc551 GraphDef subgraph_def; in TEST() local
556 cluster, graph_def, &subgraph_def)); in TEST()
557 EXPECT_EQ(3, subgraph_def.node_size()); in TEST()
560 subgraph_def.Clear(); in TEST()
565 cluster, graph_def, &subgraph_def)); in TEST()
566 EXPECT_EQ(6, subgraph_def.node_size()); in TEST()
569 subgraph_def.Clear(); in TEST()
574 cluster, graph_def, &subgraph_def)); in TEST()
575 EXPECT_EQ(10, subgraph_def.node_size()); in TEST()
578 subgraph_def.Clear(); in TEST()
[all …]
Dremote_fused_graph_execute_utils.cc607 const string& executor_name, const GraphDef& subgraph_def, in BuildRemoteFusedGraphExecuteInfo() argument
619 *execute_info->mutable_remote_graph() = subgraph_def; in BuildRemoteFusedGraphExecuteInfo()
625 GetOutputTensorShapeType(subgraph_def, input, &dt, &shape); in BuildRemoteFusedGraphExecuteInfo()
639 << "No shape type found for " << input << DumpGraphDef(subgraph_def); in BuildRemoteFusedGraphExecuteInfo()
649 GetOutputTensorShapeType(subgraph_def, output, &dt, &shape); in BuildRemoteFusedGraphExecuteInfo()
665 << "No shape type found for " << output << DumpGraphDef(subgraph_def); in BuildRemoteFusedGraphExecuteInfo()
677 const GraphDef& subgraph_def, const std::vector<string>& inputs, in BuildRemoteFusedGraphExecuteOpNode() argument
688 executor_name, subgraph_def, inputs, outputs, require_shape_type, in BuildRemoteFusedGraphExecuteOpNode()
830 GraphDef* subgraph_def) { in BuildClusterSubgraphDef() argument
846 graph.ToGraphDef(subgraph_def); in BuildClusterSubgraphDef()
[all …]
Dremote_fused_graph_execute_utils.h176 const string& executor_name, const GraphDef& subgraph_def,
185 const GraphDef& subgraph_def, const std::vector<string>& inputs,
204 GraphDef* subgraph_def);