Home
last modified time | relevance | path

Searched refs:fused_conv_2d (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_test.cc765 Tensor fused_conv_2d; in VerifyBiasAddTensorsNear() local
768 run_fused(image, filter, bias, &fused_conv_2d); in VerifyBiasAddTensorsNear()
770 ASSERT_EQ(conv_2d.dtype(), fused_conv_2d.dtype()); in VerifyBiasAddTensorsNear()
771 ASSERT_EQ(conv_2d.shape(), fused_conv_2d.shape()); in VerifyBiasAddTensorsNear()
778 test::ExpectClose(conv_2d, fused_conv_2d, /*atol=*/1e-4); in VerifyBiasAddTensorsNear()
780 test::ExpectClose(conv_2d, fused_conv_2d, /*atol=*/1e-6); in VerifyBiasAddTensorsNear()
815 Tensor fused_conv_2d; in VerifyFusedBatchNormTensorsNear() local
818 run_fused(image, filter, scale, offset, mean, variance, &fused_conv_2d); in VerifyFusedBatchNormTensorsNear()
820 ASSERT_EQ(conv_2d.dtype(), fused_conv_2d.dtype()); in VerifyFusedBatchNormTensorsNear()
821 ASSERT_EQ(conv_2d.shape(), fused_conv_2d.shape()); in VerifyFusedBatchNormTensorsNear()
[all …]
Dmkl_fused_ops_test.cc110 Tensor fused_conv_2d; in VerifyBiasAddTensorsClose() local
113 run_fused(image, filter, bias, &fused_conv_2d); in VerifyBiasAddTensorsClose()
115 ASSERT_EQ(conv_2d.dtype(), fused_conv_2d.dtype()); in VerifyBiasAddTensorsClose()
116 ASSERT_EQ(conv_2d.shape(), fused_conv_2d.shape()); in VerifyBiasAddTensorsClose()
118 test::ExpectClose(conv_2d, fused_conv_2d); in VerifyBiasAddTensorsClose()