| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling_test.cc | 188 XlaOp out_backprop = ConstantR4FromArray4D<float>(&builder, {{{{1.}}}}); in XLA_TEST_F() local
192 AvgPoolGrad(out_backprop, {1, 1, 3, 3}, kernel_size, stride, in XLA_TEST_F()
205 XlaOp out_backprop = in XLA_TEST_F() local
210 AvgPoolGrad(out_backprop, {1, 1, 3, 3}, kernel_size, stride, in XLA_TEST_F()
223 XlaOp out_backprop = in XLA_TEST_F() local
228 AvgPoolGrad(out_backprop, {1, 1, 3, 3}, kernel_size, stride, {{1, 1}, {1, 1}}, in XLA_TEST_F()
240 XlaOp out_backprop = in XLA_TEST_F() local
245 AvgPoolGrad(out_backprop, {1, 1, 3, 3}, kernel_size, stride, {{1, 1}, {1, 1}}, in XLA_TEST_F()
255 XlaOp out_backprop = in XLA_TEST_F() local
263 AvgPoolGrad(out_backprop, {1, 1, 3, 3}, kernel_size, stride, {{1, 1}, {1, 1}}, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops.cc | 222 const Tensor& out_backprop = context->input(2); in Compute() local
238 batch == out_backprop.dim_size(0) && in Compute()
239 out_rows == out_backprop.dim_size(1) && in Compute()
240 out_cols == out_backprop.dim_size(2) && in Compute()
241 depth == out_backprop.dim_size(3), in Compute()
257 filter.tensor<T, 3>(), out_backprop.tensor<T, 4>(), stride_rows, in Compute()
273 typename TTypes<T, 4>::ConstTensor out_backprop, in operator ()()
285 const int output_rows = out_backprop.dimension(1); in operator ()()
286 const int output_cols = out_backprop.dimension(2); in operator ()()
322 out_backprop(b, h_out, w_out, d); in operator ()()
[all …]
|
| D | avgpooling_op.cc | 278 const Tensor& out_backprop = context->input(1); in Compute() local
286 OP_REQUIRES(context, out_backprop.dims() == 4, in Compute()
288 const int64 out_backprop_batch = out_backprop.dim_size(0); in Compute()
289 const int64 out_backprop_rows = out_backprop.dim_size(1); in Compute()
290 const int64 out_backprop_cols = out_backprop.dim_size(2); in Compute()
291 const int64 out_backprop_depth = out_backprop.dim_size(3); in Compute()
331 const T* out_backprop_ptr = out_backprop.flat<T>().data(); in Compute()
440 const Tensor& out_backprop = context->input(1); in Compute() local
448 OP_REQUIRES(context, out_backprop.dims() == 4, in Compute()
466 /*explicit_paddings=*/{}, data_format_, nullptr, nullptr, out_backprop, in Compute()
[all …]
|
| D | depthwise_conv_grad_op.cc | 60 const Tensor& out_backprop = context->input(2); \
68 context, out_backprop.dims() == 4, \
72 context, batch == out_backprop.dim_size(0), \
90 GetTensorDim(out_backprop.shape(), data_format_, 'H'); \
97 GetTensorDim(out_backprop.shape(), data_format_, 'W'); \
109 GetTensorDim(out_backprop.shape(), data_format_, 'C'); \
195 const T* out_backprop, T* buffer) { in CopyOutputBackpropRegion() argument
236 out_backprop + (out_r * args.out_cols + out_c) * args.out_depth; in CopyOutputBackpropRegion()
387 const T* out_backprop, const T* depthwise_filter, in operator ()()
417 auto shard = [&ctx, &args, &out_backprop, &filter_data, &in_backprop]( in operator ()()
[all …]
|
| D | pooling_ops_3d.cc | 205 const Tensor& tensor_out, const Tensor& out_backprop, in launch()
212 for (int64 p = 0; p < out_backprop.dim_size(3); ++p) { in launch()
228 for (int64 r = 0; r < out_backprop.dim_size(2); ++r) { in launch()
233 for (int64 c = 0; c < out_backprop.dim_size(1); ++c) { in launch()
248 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
277 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
338 const Tensor& out_backprop = context->input(2); in Compute() local
343 OP_REQUIRES(context, out_backprop.dims() == 5, in Compute()
365 context, tensor_in, tensor_out, out_backprop, window, stride, out, in Compute()
380 const Tensor& out_backprop, in launch()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 212 typename TTypes<T, 4>::ConstTensor out_backprop, in operator ()()
224 const int output_rows = out_backprop.dimension(1); in operator ()()
225 const int output_cols = out_backprop.dimension(2); in operator ()()
243 input.data(), filter.data(), out_backprop.data(), batch, input_rows, in operator ()()
254 typename TTypes<T, 4>::ConstTensor out_backprop, in operator ()()
266 const int output_rows = out_backprop.dimension(1); in operator ()()
267 const int output_cols = out_backprop.dimension(2); in operator ()()
285 input.data(), filter.data(), out_backprop.data(), batch, input_rows, in operator ()()
|
| D | depthwise_conv_op_gpu.h | 808 const DepthwiseArgs args, const T* __restrict__ out_backprop,
861 sum += ldg(out_backprop + out_backprop_offset +
878 const DepthwiseArgs args, const T* __restrict__ out_backprop,
939 sum += ldg(out_backprop + out_backprop_offset) *
955 const T* out_backprop,
979 args, out_backprop, filter, in_backprop, num_in_backprop));
986 const T* out_backprop,
995 ctx, args, out_backprop, filter, in_backprop, data_format);
1000 ctx, args, out_backprop, filter, in_backprop, data_format);
1004 ctx, args, out_backprop, filter, in_backprop, data_format);
[all …]
|
| D | conv_grad_ops.h | 175 const Tensor& out_backprop, const Tensor& filter,
185 const Tensor& out_backprop, const Tensor& input,
206 const Tensor& out_backprop, const Tensor& input,
|
| D | batch_norm_op.h | 79 typename TTypes<T, 4>::ConstTensor out_backprop, in operator()
114 db.device(d) = out_backprop.reshape(rest_by_depth).sum(reduction_axis); in operator()
120 scratch2.device(d) = (out_backprop.reshape(rest_by_depth) * in operator()
127 out_backprop.reshape(rest_by_depth) * ((scratch1 * gamma) in operator()
135 out_backprop.reshape(rest_by_depth) * in operator()
|
| D | conv_grad_input_ops.h | 112 const Tensor& out_backprop, const Tensor& filter, in operator()
134 filter_shape, out_backprop.shape(), dilations, strides, in operator()
178 auto out_backprop_t = out_backprop.tensor<T, 4>(); in operator()
207 const Tensor& out_backprop, const Tensor& filter,
213 launcher(ctx, use_cudnn, cudnn_use_autotune, out_backprop, filter,
422 const Tensor& out_backprop = context->input(2);
427 out_backprop.shape(),
449 << " out_backprop: " << out_backprop.shape().DebugString()
454 launch(context, use_cudnn_, cudnn_use_autotune_, out_backprop, filter,
519 const Tensor& out_backprop = context->input(2);
[all …]
|
| D | conv_grad_filter_ops.cc | 109 const Tensor& out_backprop, const Tensor& input, in operator ()()
127 filter_shape, out_backprop.shape(), dilations, strides, in operator ()()
160 auto out_backprop_t = out_backprop.tensor<T, 4>(); in operator ()()
322 const Tensor& out_backprop = context->input(2); in Compute() local
357 << " out_backprop: " << out_backprop.shape().DebugString() in Compute()
361 launcher_(context, use_cudnn_, cudnn_use_autotune_, out_backprop, input, in Compute()
430 const Tensor& out_backprop = context->input(2); in Compute() local
446 filter_shape, out_backprop.shape(), dilations_, strides_, padding_, in Compute()
483 filter_backprop->tensor<T, 4>(), out_backprop.tensor<T, 4>(), in Compute()
542 const T* out_backprop_data = out_backprop.template flat<T>().data(); in Compute()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | layout_optimization_layout_assignment_gpu_cc_70.mlir | 70 %out_backprop: tensor<1x28x28x64xf32>
75 %0 = "tf.Conv2DBackpropFilter"(%input, %filter_size, %out_backprop)
90 %out_backprop: tensor<1x64x28x28xf16>
95 %0 = "tf.Conv2DBackpropFilter"(%input, %filter_size, %out_backprop)
110 %out_backprop: tensor<1x28x28x64xf32>
115 %0 = "tf.Conv2DBackpropInput"(%input_size, %filter, %out_backprop)
130 %out_backprop: tensor<1x64x28x28xf16>
135 %0 = "tf.Conv2DBackpropInput"(%input_size, %filter, %out_backprop)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | DepthwiseConv2dNativeBackpropFilter.pbtxt | 12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
|
| D | DepthwiseConv2dNativeBackpropInput.pbtxt | 12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
|
| D | BiasAddGrad.pbtxt | 4 name: "out_backprop"
50 name: "out_backprop"
98 name: "out_backprop"
147 name: "out_backprop"
|
| D | Conv3DBackpropInputV2.pbtxt | 12 name: "out_backprop"
57 name: "out_backprop"
115 name: "out_backprop"
188 name: "out_backprop"
|
| D | Conv2DBackpropInput.pbtxt | 12 name: "out_backprop"
75 name: "out_backprop"
151 name: "out_backprop"
228 name: "out_backprop"
314 name: "out_backprop"
|
| D | Dilation2DBackpropFilter.pbtxt | 12 name: "out_backprop"
70 name: "out_backprop"
130 name: "out_backprop"
191 name: "out_backprop"
|
| D | Dilation2DBackpropInput.pbtxt | 12 name: "out_backprop"
70 name: "out_backprop"
130 name: "out_backprop"
191 name: "out_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DepthwiseConv2dNativeBackpropInput.pbtxt | 12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
|
| D | DepthwiseConv2dNativeBackpropFilter.pbtxt | 12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
|
| D | BiasAddGrad.pbtxt | 4 name: "out_backprop"
50 name: "out_backprop"
98 name: "out_backprop"
147 name: "out_backprop"
|
| D | Conv3DBackpropInputV2.pbtxt | 12 name: "out_backprop"
57 name: "out_backprop"
115 name: "out_backprop"
188 name: "out_backprop"
|
| D | Conv2DBackpropInput.pbtxt | 12 name: "out_backprop"
75 name: "out_backprop"
151 name: "out_backprop"
228 name: "out_backprop"
314 name: "out_backprop"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BiasAddGrad.pbtxt | 4 name: "out_backprop"
12 1-D with size the feature dimension of `out_backprop`.
29 It accumulates all the values from out_backprop into the feature dimension.
|