Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc80 std::vector<Tensor> fused_outputs; in TestFoldBatchNormsConv2D() local
81 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldBatchNormsConv2D()
83 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldBatchNormsConv2D()
131 std::vector<Tensor> fused_outputs; in TestFoldBatchNormsDepthwiseConv2dNative() local
132 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldBatchNormsDepthwiseConv2dNative()
134 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldBatchNormsDepthwiseConv2dNative()
191 std::vector<Tensor> fused_outputs; in TestFoldBatchNormsConv2DShared() local
193 fused_session->Run({}, {"output", "output_2"}, {}, &fused_outputs)); in TestFoldBatchNormsConv2DShared()
195 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldBatchNormsConv2DShared()
196 test::ExpectTensorNear<float>(original_outputs[1], fused_outputs[1], 1e-5); in TestFoldBatchNormsConv2DShared()
[all …]
Dfuse_convolutions_test.cc89 std::vector<Tensor> fused_outputs; in TestFuseResizePadAndConv() local
90 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFuseResizePadAndConv()
92 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFuseResizePadAndConv()
139 std::vector<Tensor> fused_outputs; in TestFuseResizeAndConv() local
140 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFuseResizeAndConv()
142 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFuseResizeAndConv()
191 std::vector<Tensor> fused_outputs; in TestFusePadAndConv() local
192 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFusePadAndConv()
194 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc114 std::vector<Tensor> fused_outputs; in TestFoldOldBatchNorms() local
115 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldOldBatchNorms()
117 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldOldBatchNorms()
192 std::vector<Tensor> fused_outputs; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
193 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
195 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
269 std::vector<Tensor> fused_outputs; in TestFoldFusedBatchNorms() local
270 TF_ASSERT_OK(fused_session->Run({}, {"output"}, {}, &fused_outputs)); in TestFoldFusedBatchNorms()
272 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 2e-5); in TestFoldFusedBatchNorms()
346 std::vector<Tensor> fused_outputs; in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]