Home
last modified time | relevance | path

Searched refs:original_outputs (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc71 std::vector<Tensor> original_outputs; in TestFoldBatchNormsConv2D() local
72 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsConv2D()
83 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldBatchNormsConv2D()
122 std::vector<Tensor> original_outputs; in TestFoldBatchNormsDepthwiseConv2dNative() local
123 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldBatchNormsDepthwiseConv2dNative()
134 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldBatchNormsDepthwiseConv2dNative()
181 std::vector<Tensor> original_outputs; in TestFoldBatchNormsConv2DShared() local
183 &original_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.cc80 std::vector<Tensor> original_outputs; in TestFuseResizePadAndConv() local
81 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFuseResizePadAndConv()
92 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFuseResizePadAndConv()
130 std::vector<Tensor> original_outputs; in TestFuseResizeAndConv() local
131 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFuseResizeAndConv()
142 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFuseResizeAndConv()
182 std::vector<Tensor> original_outputs; in TestFusePadAndConv() local
183 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFusePadAndConv()
194 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc105 std::vector<Tensor> original_outputs; in TestFoldOldBatchNorms() local
106 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldOldBatchNorms()
117 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldOldBatchNorms()
183 std::vector<Tensor> original_outputs; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
184 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
195 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 1e-5); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
260 std::vector<Tensor> original_outputs; in TestFoldFusedBatchNorms() local
261 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFoldFusedBatchNorms()
272 test::ExpectTensorNear<float>(original_outputs[0], fused_outputs[0], 2e-5); in TestFoldFusedBatchNorms()
337 std::vector<Tensor> original_outputs; in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]
Dround_weights_test.cc67 std::vector<Tensor> original_outputs; in TestRoundWeights() local
68 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestRoundWeights()
79 test::ExpectTensorNear<float>(original_outputs[0], rounded_outputs[0], 0.5); in TestRoundWeights()
Dflatten_atrous_test.cc86 std::vector<Tensor> original_outputs; in TestFlattenAtrousConv() local
87 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestFlattenAtrousConv()
108 test::ExpectTensorNear<float>(original_outputs[0], modified_outputs[0], in TestFlattenAtrousConv()
Dfreeze_requantization_ranges_test.cc141 std::vector<Tensor> original_outputs; in TestFreezeRequantizationRanges() local
143 original_session->Run({}, {"dequantize_op"}, {}, &original_outputs)); in TestFreezeRequantizationRanges()
151 ASSERT_EQ(original_outputs.size(), frozen_outputs.size()); in TestFreezeRequantizationRanges()
153 test::ExpectTensorNear<float>(original_outputs[0], frozen_outputs[0], 0.5); in TestFreezeRequantizationRanges()
Dquantize_weights_test.cc99 std::vector<Tensor> original_outputs; in TestQuantizeWeights() local
100 TF_ASSERT_OK(original_session->Run({}, {"output"}, {}, &original_outputs)); in TestQuantizeWeights()
110 test::ExpectTensorNear<float>(original_outputs[0], quantized_outputs[0], in TestQuantizeWeights()
/external/tensorflow/tensorflow/python/saved_model/
Dload.py321 original_outputs = coder.decode_proto(proto.output_signature)
327 …concrete_function._func_graph.structured_outputs = original_outputs # pylint: disable=protected-a…