Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils_test.cc70 RemoteFusedGraphExecuteOpTestUtils::BuildMultipleAddGraph(&graph_def_)); in SetUp()
90 GraphDef graph_def_with_shapetype = graph_def_; in FuseByInOut()
103 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByNodes()
110 graph_def_, inputs_, outputs_, "remote_fused_graph_node_names", in FuseByOpTypes()
117 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME0, in FuseByExecutor0()
123 graph_def_, inputs_, outputs_, REMOTE_FUSED_EXECUTOR_NAME1, in FuseByExecutor1()
129 input_tensors_, /*dry_run_inference=*/true, &graph_def_); in BuildAndAddTensorShape()
137 &graph_def_); in PlaceRemoteGraphArguments()
143 graph_def_, input_tensors_, &graph_def_); in FuseByPlacedArguments()
144 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::__anond0e737360111::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::__anonedc01a510111::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/core/common_runtime/
Dgraph_constructor.cc438 graph_def_(std::move(graph_def)), in NodeDefMovingGraphConstructor()
439 is_consumed_(graph_def_.node_size(), false) {} in NodeDefMovingGraphConstructor()
442 size_t node_def_count() const override { return graph_def_.node().size(); } in node_def_count()
446 return graph_def_.node(i); in get_node_def()
451 return std::move(*graph_def_.mutable_node(i)); in consume_node_def()
453 const VersionDef* versions() const override { return &graph_def_.versions(); } in versions()
455 return &graph_def_.library(); in library()
458 GraphDef graph_def_; member in tensorflow::__anond78e089a0111::NodeDefMovingGraphConstructor
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.cc287 SortByExecutionOrder(graph_def, &graph_def_).IgnoreError(); in GraphMatcher()
288 MapNamesToNodes(graph_def_, &node_map_); in GraphMatcher()
294 for (const NodeDef& node : graph_def_.node()) { in GetOpTypeMatches()
Dtransform_utils.h178 GraphDef graph_def_; variable