Home
last modified time | relevance | path

Searched refs:input_backward (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc40 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols); in TEST() local
46 input_backward.setRandom(); in TEST()
48 input_backward = SpatialConvolutionBackwardInput(kernel, output_backward, in TEST()
51 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
52 EXPECT_EQ(input_backward.dimension(1), input_rows); in TEST()
53 EXPECT_EQ(input_backward.dimension(2), input_cols); in TEST()
72 EigenApprox(input_backward(id, i, j), expected); in TEST()
89 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows, in TEST() local
98 input_backward.setRandom(); in TEST()
100 input_backward = SpatialConvolutionBackwardInput(kernel, output_backward, in TEST()
[all …]
Deigen_backward_cuboid_convolutions_test.cc43 Tensor<float, 4> input_backward(input_depth, input_planes, input_rows, in TEST() local
52 input_backward.setRandom(); in TEST()
54 input_backward = CuboidConvolutionBackwardInput( in TEST()
57 EXPECT_EQ(input_backward.dimension(3), input_cols); in TEST()
58 EXPECT_EQ(input_backward.dimension(2), input_rows); in TEST()
59 EXPECT_EQ(input_backward.dimension(1), input_planes); in TEST()
60 EXPECT_EQ(input_backward.dimension(0), input_depth); in TEST()
85 EigenApprox(input_backward(id, i, j, k), expected); in TEST()
106 Tensor<float, 4, RowMajor> input_backward(input_cols, input_rows, in TEST() local
115 input_backward.setRandom(); in TEST()
[all …]
Dconv_3d.h57 typename TTypes<T, 5>::Tensor input_backward,
62 input_backward.device(d) = Eigen::CuboidConvolutionBackwardInput(
64 input_backward.dimension(3), // input_planes
65 input_backward.dimension(2), // input_rows
66 input_backward.dimension(1), // input_cols
Dconv_2d.h107 void operator()(const Device& d, typename TTypes<T, 4>::Tensor input_backward,
113 input_backward.device(d) = Eigen::SpatialConvolutionBackwardInput(
114 filter, output_backward, input_backward.dimension(2),
115 input_backward.dimension(1), col_stride, row_stride, col_dilation,
124 typename TTypes<T, 4>::Tensor input_backward,
130 To32Bit(input_backward).device(d) = Eigen::SpatialConvolutionBackwardInput(
131 To32Bit(filter), To32Bit(output_backward), input_backward.dimension(2),
132 input_backward.dimension(1), col_stride, row_stride, col_dilation,
139 void operator()(const Device& d, typename TTypes<T, 4>::Tensor input_backward,
152 input_backward.device(d) =
[all …]
Deigen_benchmark.h102 InputBackward input_backward(input_backward_data, input_dims); in SpatialConvolutionBackwardInput()
105 input_backward.device(device_) = Eigen::SpatialConvolutionBackwardInput( in SpatialConvolutionBackwardInput()
107 tensorflow::testing::DoNotOptimize(input_backward); in SpatialConvolutionBackwardInput()
236 InputBackward input_backward(input_backward_data, input_dims); in CuboidConvolutionBackwardInput()
239 input_backward.device(device_) = Eigen::CuboidConvolutionBackwardInput( in CuboidConvolutionBackwardInput()
241 tensorflow::testing::DoNotOptimize(input_backward); in CuboidConvolutionBackwardInput()
Dconv_grad_input_ops.h223 typename TTypes<T, 4>::Tensor input_backward,
236 typename TTypes<float, 4>::Tensor input_backward,
242 auto batch = input_backward.dimension(0);
243 auto in_depth = input_backward.dimension(3);
275 auto input_ptr = input_backward.data();
Dconv_grad_filter_ops.cc195 typename TTypes<T, 4>::ConstTensor input_backward, in operator ()()