| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions_test.cc | 42 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols); in TEST() local
44 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
48 input_backward = SpatialConvolutionBackwardInput(kernel, output_backward, in TEST()
66 expected += output_backward(od, output_i, output_j) * in TEST()
93 Tensor<float, 3, RowMajor> output_backward(output_cols, output_rows, in TEST() local
96 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
100 input_backward = SpatialConvolutionBackwardInput(kernel, output_backward, in TEST()
118 expected += output_backward(output_j, output_i, od) * in TEST()
143 Tensor<float, 3> output_backward(output_depth, output_rows, output_cols); in TEST() local
145 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 47 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST() local
50 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
55 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
78 output_backward(od, output_i, output_j, output_k) * in TEST()
110 Tensor<float, 4, RowMajor> output_backward(output_cols, output_rows, in TEST() local
113 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
118 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
141 output_backward(output_k, output_j, output_i, od) * in TEST()
173 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST() local
176 output_backward = output_backward.constant(11.0f) + output_backward.random(); in TEST()
[all …]
|
| D | eigen_backward_spatial_convolutions.h | 100 const Kernel& kernel, const OutputBackward& output_backward,
113 out(output_backward);
145 ? output_backward.dimension(1)
146 : output_backward.dimension(NumDims - 2);
148 ? output_backward.dimension(2)
149 : output_backward.dimension(NumDims - 3);
284 output_backward
292 output_backward
400 const Input& input, const OutputBackward& output_backward,
413 out(output_backward);
[all …]
|
| D | conv_3d.h | 59 typename TTypes<T, 5>::ConstTensor output_backward,
63 filter, output_backward,
76 typename TTypes<T, 5>::ConstTensor output_backward,
80 input, output_backward,
|
| D | eigen_benchmark.h | 101 OutputBackward output_backward(output_backward_data, output_dims); in SpatialConvolutionBackwardInput()
106 filter, output_backward, input_rows, input_cols); in SpatialConvolutionBackwardInput()
140 OutputBackward output_backward(output_backward_data, input_dims); in SpatialConvolutionBackwardKernel()
145 input, output_backward, filter_rows, filter_cols); in SpatialConvolutionBackwardKernel()
235 OutputBackward output_backward(output_backward_data, output_dims); in CuboidConvolutionBackwardInput()
240 filter, output_backward, input_planes, input_rows, input_cols); in CuboidConvolutionBackwardInput()
276 OutputBackward output_backward(output_backward_data, output_dims); in CuboidConvolutionBackwardKernel()
281 input, output_backward, filter_planes, filter_rows, filter_cols); in CuboidConvolutionBackwardKernel()
|
| D | conv_2d.h | 109 typename TTypes<T, 4>::ConstTensor output_backward,
114 filter, output_backward, input_backward.dimension(2),
126 typename TTypes<T, 4>::ConstTensor output_backward,
131 To32Bit(filter), To32Bit(output_backward), input_backward.dimension(2),
141 typename TTypes<T, 4>::ConstTensor output_backward,
154 filter, output_backward, padded_cols, padded_rows, col_stride,
169 typename TTypes<T, 4>::ConstTensor output_backward,
177 To32Bit(filter), To32Bit(output_backward), padded_cols, padded_rows,
|
| D | eigen_backward_cuboid_convolutions.h | 88 const Kernel& kernel, const OutputBackward& output_backward,
103 out(output_backward);
280 .contract(output_backward
289 output_backward
390 const Input& input, const OutputBackward& output_backward,
404 out(output_backward);
578 output_backward.shuffle(output_backward_shuffle).eval();
|
| D | conv_grad_input_ops.h | 225 typename TTypes<T, 4>::ConstTensor output_backward,
238 typename TTypes<float, 4>::ConstTensor output_backward,
244 auto out_depth = output_backward.dimension(3);
277 auto output_ptr = output_backward.data();
|
| D | conv_grad_filter_ops.cc | 197 typename TTypes<T, 4>::ConstTensor output_backward, in operator ()()
|