Searched refs:kernel_backward (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions_test.cc | 774 const Tensor<float, 4> kernel_backward = in TEST() local
782 EXPECT_EQ(kernel_backward.dimension(0), output_depth); in TEST()
783 EXPECT_EQ(kernel_backward.dimension(1), input_depth); in TEST()
784 EXPECT_EQ(kernel_backward.dimension(2), patch_rows); in TEST()
785 EXPECT_EQ(kernel_backward.dimension(3), patch_cols); in TEST()
787 eigen_assert(dimensions_match(kernel_backward.dimensions(), in TEST()
789 for (ptrdiff_t i = 0; i < kernel_backward.dimensions().TotalSize(); ++i) { in TEST()
790 EigenApprox(kernel_backward.data()[i], expected_kernel_backward.data()[i]); in TEST()
826 const Tensor<float, 4, RowMajor> kernel_backward = in TEST() local
830 EXPECT_EQ(kernel_backward.dimension(0), patch_cols); in TEST()
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 580 Tensor<float, 5> kernel_backward(output_depth, input_depth, patch_planes, in TEST() local
587 kernel_backward.setRandom(); in TEST()
589 kernel_backward = CuboidConvolutionBackwardKernel( in TEST()
592 EXPECT_EQ(kernel_backward.dimension(0), output_depth); in TEST()
593 EXPECT_EQ(kernel_backward.dimension(1), input_depth); in TEST()
594 EXPECT_EQ(kernel_backward.dimension(2), patch_planes); in TEST()
595 EXPECT_EQ(kernel_backward.dimension(3), patch_rows); in TEST()
596 EXPECT_EQ(kernel_backward.dimension(4), patch_cols); in TEST()
622 EigenApprox(kernel_backward(od, id, p, r, c), expected); in TEST()
647 Tensor<float, 5, RowMajor> kernel_backward( in TEST() local
[all …]
|