Home
last modified time | relevance | path

Searched refs:out_backprop (Results 1 – 25 of 95) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling_test.cc188 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/
Ddilation_ops.cc222 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 …]
Davgpooling_op.cc278 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 …]
Ddepthwise_conv_grad_op.cc60 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 …]
Dpooling_ops_3d.cc205 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 …]
Ddilation_ops_gpu.cu.cc212 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 ()()
Ddepthwise_conv_op_gpu.h808 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 …]
Dconv_grad_ops.h175 const Tensor& out_backprop, const Tensor& filter,
185 const Tensor& out_backprop, const Tensor& input,
206 const Tensor& out_backprop, const Tensor& input,
Dbatch_norm_op.h79 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()
Dconv_grad_input_ops.h112 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 …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlayout_optimization_layout_assignment_gpu_cc_70.mlir70 %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_v2/
DDepthwiseConv2dNativeBackpropInput.pbtxt12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
DDepthwiseConv2dNativeBackpropFilter.pbtxt12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
DBiasAddGrad.pbtxt4 name: "out_backprop"
50 name: "out_backprop"
98 name: "out_backprop"
147 name: "out_backprop"
DConv3DBackpropInputV2.pbtxt12 name: "out_backprop"
57 name: "out_backprop"
115 name: "out_backprop"
188 name: "out_backprop"
DConv2DBackpropInput.pbtxt12 name: "out_backprop"
75 name: "out_backprop"
151 name: "out_backprop"
228 name: "out_backprop"
314 name: "out_backprop"
DDilation2DBackpropInput.pbtxt12 name: "out_backprop"
70 name: "out_backprop"
130 name: "out_backprop"
191 name: "out_backprop"
DDilation2DBackpropFilter.pbtxt12 name: "out_backprop"
70 name: "out_backprop"
130 name: "out_backprop"
191 name: "out_backprop"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DDepthwiseConv2dNativeBackpropFilter.pbtxt12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
DDepthwiseConv2dNativeBackpropInput.pbtxt12 name: "out_backprop"
55 name: "out_backprop"
111 name: "out_backprop"
180 name: "out_backprop"
250 name: "out_backprop"
DBiasAddGrad.pbtxt4 name: "out_backprop"
50 name: "out_backprop"
98 name: "out_backprop"
147 name: "out_backprop"
DConv3DBackpropInputV2.pbtxt12 name: "out_backprop"
57 name: "out_backprop"
115 name: "out_backprop"
188 name: "out_backprop"
DConv2DBackpropInput.pbtxt12 name: "out_backprop"
75 name: "out_backprop"
151 name: "out_backprop"
228 name: "out_backprop"
314 name: "out_backprop"
DDilation2DBackpropFilter.pbtxt12 name: "out_backprop"
70 name: "out_backprop"
130 name: "out_backprop"
191 name: "out_backprop"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BiasAddGrad.pbtxt4 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.

1234