Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_graph_optimization_pass_test.cc111 void verifyGraph(const GraphDef& original_graph_def, bool changed = false) { in verifyGraph() argument
120 ::testing::EquivToProto(original_graph_def)))); in verifyGraph()
124 ::testing::EquivToProto(original_graph_def))); in verifyGraph()
141 GraphDef original_graph_def; in TEST_F() local
142 graph_->ToGraphDef(&original_graph_def); in TEST_F()
148 verifyGraph(original_graph_def); in TEST_F()
156 GraphDef original_graph_def; in TEST_F() local
157 graph_->ToGraphDef(&original_graph_def); in TEST_F()
163 verifyGraph(original_graph_def); in TEST_F()
170 GraphDef original_graph_def; in TEST_F() local
[all …]
/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 …]
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()
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 …]
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/python/tools/
Doptimize_for_inference_test.py160 original_graph_def = sess.graph_def
163 original_graph_def)
221 original_graph_def = sess.graph_def
224 original_graph_def)
281 original_graph_def = sess.graph_def
284 original_graph_def)
311 original_graph_def = sess.graph_def
314 original_graph_def, ["output"])
341 original_graph_def = sess.graph_def
344 original_graph_def, ["output"])
[all …]
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py690 original_graph_def = self._MakeGraphDef("""
699 original_graph_def, return_elements=["B"], name="imported_graph")
707 original_graph_def = self._MakeGraphDef("""
718 a, b = importer.import_graph_def(original_graph_def,
727 original_graph_def = self._MakeGraphDef("""
738 a, b = importer.import_graph_def(original_graph_def,
749 original_graph_def = self._MakeGraphDef("""
770 a, b = importer.import_graph_def(original_graph_def,
788 a, b = importer.import_graph_def(original_graph_def,
804 a, b = importer.import_graph_def(original_graph_def,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test_utils.cc117 const GraphDef& original_graph_def, const std::vector<string>& inputs, in FuseRemoteGraph() argument
120 original_graph_def, inputs, outputs, "remote_fused_graph_node_names", in FuseRemoteGraph()
Di_remote_fused_graph_executor.h64 virtual Status FuseRemoteGraph(const GraphDef& original_graph_def,
Dremote_fused_graph_execute_op_test_utils.h76 Status FuseRemoteGraph(const GraphDef& original_graph_def,
Dremote_fused_graph_execute_op_test.cc217 Status FuseRemoteGraph(const GraphDef& original_graph_def, in FuseRemoteGraph() argument
221 *fused_graph_def = original_graph_def; in FuseRemoteGraph()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_control_wrapper.h51 Status FuseRemoteGraph(const GraphDef& original_graph_def,
Dhexagon_control_wrapper.cc397 const GraphDef& original_graph_def, const std::vector<string>& inputs, in FuseRemoteGraph() argument
401 original_graph_def, HexagonOpsDefinitions::getInstance()); in FuseRemoteGraph()
405 original_graph_def, inputs, outputs, REMOTE_FUSED_GRAPH_NODE_NAME_PREFIX, in FuseRemoteGraph()
/external/tensorflow/tensorflow/core/protobuf/
Ddebug_event.proto208 bytes original_graph_def = 4; field