Home
last modified time | relevance | path

Searched refs:original_graph_def (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms_test.cc80 GraphDef original_graph_def; in TestFoldOldBatchNorms() local
81 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldOldBatchNorms()
100 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldOldBatchNorms()
101 original_graph_def.mutable_versions()->set_producer(8); in TestFoldOldBatchNorms()
104 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldOldBatchNorms()
109 TF_ASSERT_OK(FoldOldBatchNorms(original_graph_def, {{}, {"output"}}, in TestFoldOldBatchNorms()
164 GraphDef original_graph_def; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
165 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
178 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
179 original_graph_def.mutable_versions()->set_producer(8); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
[all …]
Dfold_batch_norms_test.cc66 GraphDef original_graph_def; in TestFoldBatchNormsConv2D() local
67 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsConv2D()
70 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsConv2D()
76 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsConv2D()
117 GraphDef original_graph_def; in TestFoldBatchNormsDepthwiseConv2dNative() local
118 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
121 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
127 FoldBatchNorms(original_graph_def, {{}, {"output"}}, &fused_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
176 GraphDef original_graph_def; in TestFoldBatchNormsConv2DShared() local
177 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFoldBatchNormsConv2DShared()
[all …]
Dfuse_convolutions_test.cc75 GraphDef original_graph_def; in TestFuseResizePadAndConv() local
76 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFuseResizePadAndConv()
79 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFuseResizePadAndConv()
84 TF_ASSERT_OK(FuseResizePadAndConv(original_graph_def, {{}, {"output"}}, in TestFuseResizePadAndConv()
125 GraphDef original_graph_def; in TestFuseResizeAndConv() local
126 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFuseResizeAndConv()
129 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFuseResizeAndConv()
134 TF_ASSERT_OK(FuseResizeAndConv(original_graph_def, {{}, {"output"}}, in TestFuseResizeAndConv()
177 GraphDef original_graph_def; in TestFusePadAndConv() local
178 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFusePadAndConv()
[all …]
Dquantize_weights_test.cc42 GraphDef* original_graph_def) { in BuildGraphDef() argument
58 TF_ASSERT_OK(root.ToGraphDef(original_graph_def)); in BuildGraphDef()
62 GraphDef original_graph_def; in TestQuantizeWeights() local
71 &original_graph_def); in TestQuantizeWeights()
78 QuantizeWeights(original_graph_def, context, &quantized_graph_def)); in TestQuantizeWeights()
98 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestQuantizeWeights()
118 GraphDef original_graph_def; in TEST_F() local
127 &original_graph_def); in TEST_F()
133 QuantizeWeights(original_graph_def, context, &quantized_graph_def)); in TEST_F()
Dround_weights_test.cc62 GraphDef original_graph_def; in TestRoundWeights() local
63 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestRoundWeights()
66 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestRoundWeights()
72 RoundWeights(original_graph_def, {{}, {"output"}}, &rounded_graph_def)); in TestRoundWeights()
Dflatten_atrous_test.cc81 GraphDef original_graph_def; in TestFlattenAtrousConv() local
82 TF_ASSERT_OK(root.ToGraphDef(&original_graph_def)); in TestFlattenAtrousConv()
85 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFlattenAtrousConv()
90 TF_ASSERT_OK(FlattenAtrousConv(original_graph_def, {{}, {"output"}}, in TestFlattenAtrousConv()
Dfold_constants_lib.cc46 Status ReplaceSendRecvs(const GraphDef& original_graph_def, in ReplaceSendRecvs() argument
68 for (const NodeDef& node : original_graph_def.node()) { in ReplaceSendRecvs()
Dfold_constants_test.cc38 Status ReplaceSendRecvs(const GraphDef& original_graph_def,
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_graph_optimization_pass_test.cc129 void verifyGraph(const GraphDef& original_graph_def, bool changed = false) { in verifyGraph() argument
138 ::testing::EquivToProto(original_graph_def)))); in verifyGraph()
142 ::testing::EquivToProto(original_graph_def))); in verifyGraph()
159 GraphDef original_graph_def; in TEST_F() local
160 graph_->ToGraphDef(&original_graph_def); in TEST_F()
166 verifyGraph(original_graph_def); in TEST_F()
174 GraphDef original_graph_def; in TEST_F() local
175 graph_->ToGraphDef(&original_graph_def); in TEST_F()
183 verifyGraph(original_graph_def); in TEST_F()
189 GraphDef original_graph_def; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_test.py156 original_graph_def = sess.graph_def
159 original_graph_def)
217 original_graph_def = sess.graph_def
220 original_graph_def)
277 original_graph_def = sess.graph_def
280 original_graph_def)
307 original_graph_def = sess.graph_def
310 original_graph_def, ["output"])
337 original_graph_def = sess.graph_def
340 original_graph_def, ["output"])
[all …]
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py686 original_graph_def = self._MakeGraphDef("""
695 original_graph_def, return_elements=["B"], name="imported_graph")
703 original_graph_def = self._MakeGraphDef("""
714 a, b = importer.import_graph_def(original_graph_def,
723 original_graph_def = self._MakeGraphDef("""
734 a, b = importer.import_graph_def(original_graph_def,
745 original_graph_def = self._MakeGraphDef("""
766 a, b = importer.import_graph_def(original_graph_def,
784 a, b = importer.import_graph_def(original_graph_def,
800 a, b = importer.import_graph_def(original_graph_def,
[all …]
/external/tensorflow/tensorflow/core/tfrt/utils/
Dtfrt_graph_execution_state.h91 const GraphDef* original_graph_def() const { in original_graph_def() function
93 return graph_execution_state_->original_graph_def(); in original_graph_def()
Dtfrt_graph_execution_state_test.cc756 ASSERT_NE(graph_execution_state->original_graph_def(), nullptr); in TEST_F()
757 CompareGraphs(expected, *graph_execution_state->original_graph_def()); in TEST_F()
Dtfrt_graph_execution_state.cc493 auto* graph_def = graph_execution_state_->original_graph_def(); in Extend()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.h152 GraphDef* original_graph_def() { return original_graph_def_.get(); } in original_graph_def() function
/external/tensorflow/tensorflow/core/protobuf/
Ddebug_event.proto208 bytes original_graph_def = 4; field