Home
last modified time | relevance | path

Searched refs:original_graph_def (Results 1 – 18 of 18) 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/python/tools/
Doptimize_for_inference_test.py160 original_graph_def = sess.graph_def
163 original_graph_def)
212 original_graph_def = sess.graph_def
215 original_graph_def)
242 original_graph_def = sess.graph_def
245 original_graph_def, ["output"])
272 original_graph_def = sess.graph_def
275 original_graph_def, ["output"])
302 original_graph_def = sess.graph_def
305 original_graph_def, ["output"])
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py704 original_graph_def = self._MakeGraphDef("""
713 original_graph_def, return_elements=["B"], name="imported_graph")
721 original_graph_def = self._MakeGraphDef("""
732 a, b = importer.import_graph_def(original_graph_def,
741 original_graph_def = self._MakeGraphDef("""
752 a, b = importer.import_graph_def(original_graph_def,
763 original_graph_def = self._MakeGraphDef("""
784 a, b = importer.import_graph_def(original_graph_def,
802 a, b = importer.import_graph_def(original_graph_def,
818 a, b = importer.import_graph_def(original_graph_def,
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.h116 const GraphDef& original_graph_def,
162 const GraphDef& original_graph_def() { return original_graph_def_; } in original_graph_def() function
Ddirect_session.cc1237 execution_state_->original_graph_def().versions().producer(); in CreateExecutors()
1478 execution_state_->original_graph_def().library(), prune_options, in CreateGraphs()
1479 execution_state_->original_graph_def(), subgraph_options, in CreateGraphs()
/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.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()
Dhexagon_control_wrapper.h51 Status FuseRemoteGraph(const GraphDef& original_graph_def,