Searched refs:conv_2d (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 764 Tensor conv_2d; in VerifyBiasAddTensorsNear() local
767 run_default(image, filter, bias, &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()
814 Tensor conv_2d; in VerifyFusedBatchNormTensorsNear() local
817 run_default(image, filter, scale, offset, mean, variance, &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 …]
|
| D | mkl_fused_ops_test.cc | 109 Tensor conv_2d; in VerifyBiasAddTensorsClose() local
112 run_default(image, filter, bias, &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()
|
| D | BUILD | 256 name = "conv_2d",
257 hdrs = ["conv_2d.h"],
259 "conv_2d.h",
287 hdrs = ["conv_2d.h"],
1865 ":conv_2d",
3789 ":conv_2d",
3889 ":conv_2d",
4132 ":conv_2d",
5593 "conv_2d.h",
5660 "conv_2d.h",
|