Home
last modified time | relevance | path

Searched refs:graph_def_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc70 RemoteFusedGraphExecuteOpTestUtils::BuildMultipleAddGraph(&graph_def_)); in SetUp()
91 GraphDef graph_def_with_shapetype = graph_def_; in FuseByInOut()
104 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByNodes()
111 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByOpTypes()
118 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME0, in FuseByExecutor0()
124 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME1, in FuseByExecutor1()
130 input_tensors_, /*dry_run_inference=*/true, &graph_def_); in BuildAndAddTensorShape()
138 &graph_def_); in PlaceRemoteGraphArguments()
144 graph_def_, input_tensors_, &graph_def_); in FuseByPlacedArguments()
145 result_graph_def_ = graph_def_; in FuseByPlacedArguments()
[all …]
/external/tensorflow/tensorflow/core/grappler/inputs/
Dutils_test.cc43 auto node = graph_def_.add_node(); in SetUp()
46 TF_CHECK_OK(WriteTextProto(env_, text_graph_def_file_, graph_def_)); in SetUp()
47 TF_CHECK_OK(WriteBinaryProto(env_, binary_graph_def_file_, graph_def_)); in SetUp()
48 *meta_graph_def_.mutable_graph_def() = graph_def_; in SetUp()
61 GraphDef graph_def_; member in tensorflow::grappler::__anon995275d80111::UtilsTest
89 EXPECT_EQ(result.DebugString(), graph_def_.DebugString()); in TEST_F()
95 EXPECT_EQ(result.DebugString(), graph_def_.DebugString()); in TEST_F()
/external/tensorflow/tensorflow/core/example/
Dexample_parser_configuration_test.cc50 protobuf::TextFormat::ParseFromString(proto_string, &graph_def_); in SetUp()
52 TF_CHECK_OK(session_->Create(graph_def_)); in SetUp()
56 for (auto& node : *graph_def_.mutable_node()) { in parse_example_node()
64 GraphDef graph_def_; member in tensorflow::__anon239990b30111::ExtractExampleParserConfigurationTest
72 graph_def_, "BlarseExample/ParseExample", session_.get(), &dense_vec, in TEST_F()
87 graph_def_, "ParseExample/ParseExample", session_.get(), &dense_vec, in TEST_F()
102 graph_def_, "ParseExample/ParseExample", session_.get(), &dense_vec, in TEST_F()
112 graph_def_, "ParseExample/ParseExample", session_.get(), &dense_vec, in TEST_F()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dcluster.cc20 graph_def_ = graph_def; in SetGraphDefInfo()
21 for (const tensorflow::NodeDef& node : graph_def_->node()) { in SetGraphDefInfo()
34 for (const tensorflow::NodeDef& node : graph_def_->node()) { in FindClusterInputsAndOutputs()
Dcluster.h80 const tensorflow::GraphDef* graph_def_; /*Not owned*/ variable
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc285 SortByExecutionOrder(graph_def, &graph_def_).IgnoreError(); in GraphMatcher()
286 MapNamesToNodes(graph_def_, &node_map_); in GraphMatcher()
292 for (const NodeDef& node : graph_def_.node()) { in GetOpTypeMatches()
Dtransform_utils.h178 GraphDef graph_def_; variable