Home
last modified time | relevance | path

Searched refs:fused_session (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc78 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2D() local
79 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsConv2D()
81 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldBatchNormsConv2D()
129 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsDepthwiseConv2dNative() local
130 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsDepthwiseConv2dNative()
132 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldBatchNormsDepthwiseConv2dNative()
189 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsConv2DShared() local
190 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldBatchNormsConv2DShared()
193 fused_session->Run({}, {"output", "output_2"}, {}, &fused_outputs)); in TestFoldBatchNormsConv2DShared()
237 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldBatchNormsMatMul() local
[all …]
Dfuse_convolutions_test.cc87 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFuseResizePadAndConv() local
88 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFuseResizePadAndConv()
90 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFuseResizePadAndConv()
137 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFuseResizeAndConv() local
138 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFuseResizeAndConv()
140 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFuseResizeAndConv()
189 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFusePadAndConv() local
190 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFusePadAndConv()
192 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc112 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldOldBatchNorms() local
113 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldOldBatchNorms()
115 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldOldBatchNorms()
190 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
191 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
193 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
267 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNorms() local
268 TF_ASSERT_OK(fused_session->Create(fused_graph_def)); in TestFoldFusedBatchNorms()
270 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldFusedBatchNorms()
344 std::unique_ptr<Session> fused_session(NewSession(SessionOptions())); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]