Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc69 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2D() local
70 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsConv2D()
72 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsConv2D()
120 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsDepthwiseConv2dNative() local
121 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
123 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsDepthwiseConv2dNative()
179 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2DShared() local
180 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldBatchNormsConv2DShared()
182 TF_ASSERT_OK(original_session->Run({}, {"output", "output_2"}, {}, in TestFoldBatchNormsConv2DShared()
228 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldBatchNormsMatMul() local
[all …]
Dfuse_convolutions_test.cc78 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFuseResizePadAndConv() local
79 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFuseResizePadAndConv()
81 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFuseResizePadAndConv()
128 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFuseResizeAndConv() local
129 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFuseResizeAndConv()
131 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFuseResizeAndConv()
180 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFusePadAndConv() local
181 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFusePadAndConv()
183 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc103 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldOldBatchNorms() local
104 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldOldBatchNorms()
106 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldOldBatchNorms()
181 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
182 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
184 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
258 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNorms() local
259 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFoldFusedBatchNorms()
261 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldFusedBatchNorms()
335 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]
Dround_weights_test.cc65 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestRoundWeights() local
66 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestRoundWeights()
68 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestRoundWeights()
Dflatten_atrous_test.cc84 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFlattenAtrousConv() local
85 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestFlattenAtrousConv()
87 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFlattenAtrousConv()
Dquantize_weights_test.cc97 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestQuantizeWeights() local
98 TF_ASSERT_OK(original_session->Create(original_graph_def)); in TestQuantizeWeights()
100 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestQuantizeWeights()
Dfreeze_requantization_ranges_test.cc139 std::unique_ptr<Session> original_session(NewSession(SessionOptions())); in TestFreezeRequantizationRanges() local
140 TF_ASSERT_OK(original_session->Create(graph_def)); in TestFreezeRequantizationRanges()
143 original_session->Run({}, {"dequantize_op"}, {}, &original_outputs)); in TestFreezeRequantizationRanges()