| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 116 Tensor* in_backprop, TensorFormat data_format) { in operator()
127 const TensorShape& input_shape = in_backprop->shape(); in operator()
170 REQUIRES_32BIT(in_backprop->dim_size(0)); in operator()
171 REQUIRES_32BIT(in_backprop->dim_size(1) + padding_top + padding_bottom); in operator()
172 REQUIRES_32BIT(in_backprop->dim_size(2) + padding_left + padding_right); in operator()
173 REQUIRES_32BIT(in_backprop->dim_size(3)); in operator()
177 auto in_backprop_t = in_backprop->tensor<T, 4>(); in operator()
211 Tensor* in_backprop, TensorFormat data_format) {
215 explicit_paddings, in_backprop, data_format);
430 Tensor* in_backprop = nullptr;
[all …]
|
| D | conv_grad_input_ops.cc | 56 Tensor* in_backprop, TensorFormat data_format) { in operator ()()
60 explicit_paddings, in_backprop, data_format); in operator ()()
70 const std::vector<int64>& explicit_paddings, Tensor* in_backprop, in operator ()() argument
84 TensorShape input_shape = in_backprop->shape(); in operator ()()
145 auto c_ptr = AsDeviceMemory(in_backprop->template flat<T>().data(), in operator ()()
146 in_backprop->template flat<T>().size()); in operator ()()
178 auto c_ptr = AsDeviceMemory(in_backprop->template flat<T>().data(), in operator ()()
179 in_backprop->template flat<T>().size()); in operator ()()
546 in_backprop->tensor<T, 4>()); in operator ()()
548 *in_backprop = pre_transformed_in_backprop; in operator ()()
|
| D | dilation_ops.cc | 246 Tensor* in_backprop = nullptr; in Compute() local
248 context->allocate_output(0, input.shape(), &in_backprop)); in Compute()
259 in_backprop->tensor<T, 4>()); in Compute()
276 typename TTypes<T, 4>::Tensor in_backprop) { in operator ()()
289 in_backprop.setZero(); in operator ()()
321 in_backprop(b, h_in_max, w_in_max, d) += in operator ()()
|
| D | conv_grad_ops_3d.cc | 248 Tensor* in_backprop; in Compute() local
250 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
254 in_backprop->tensor<T, 5>(), // input_backward in Compute()
355 Tensor* in_backprop; in Compute() local
357 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
445 in_backprop->tensor<T, 5>(), // input_backward in Compute()
474 auto in_backprop_flat = in_backprop->template flat<T>(); in Compute()
1173 Tensor* in_backprop; in Compute() local
1175 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
1195 auto c_ptr = AsDeviceMemory(in_backprop->template flat<T>().data(), in Compute()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 215 typename TTypes<T, 4>::Tensor in_backprop) { in operator ()()
235 total_count, in_backprop.data())); in operator ()()
246 in_backprop.data())); in operator ()()
|
| D | depthwise_conv_grad_op.cc | 388 T* in_backprop, TensorFormat data_format) { in operator ()()
417 auto shard = [&ctx, &args, &out_backprop, &filter_data, &in_backprop]( in operator ()()
458 in_backprop + b * input_image_size); in operator ()()
475 T* in_backprop) { in DepthwiseConvBackpropInputReference() argument
520 in_backprop[in_backprop_offset] = sum; in DepthwiseConvBackpropInputReference()
628 Tensor* in_backprop = nullptr; in Compute() local
630 {0}, 0, input_shape, &in_backprop)); in Compute()
677 stride_, stride_, padding_, explicit_paddings_, in_backprop, in Compute()
684 auto in_backprop_ptr = in_backprop->template flat<T>().data(); in Compute()
|
| D | depthwise_conv_op_gpu.h | 809 const T* __restrict__ filter, T* __restrict__ in_backprop,
870 in_backprop[in_backprop_offset] = sum;
879 const T* __restrict__ filter, T* __restrict__ in_backprop,
947 in_backprop[in_backprop_offset] = sum;
956 const T* filter, T* in_backprop,
979 args, out_backprop, filter, in_backprop, num_in_backprop));
987 const T* filter, T* in_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 | dilation_ops.h | 46 typename TTypes<T, 4>::Tensor in_backprop);
|
| D | conv_grad_ops.h | 179 Tensor* in_backprop, TensorFormat data_format);
|
| D | depthwise_conv_op.h | 72 const T* out_backprop, const T* filter, T* in_backprop,
94 const T* out_backprop, const T* filter, T* in_backprop,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | Dilation2DBackpropInput.pbtxt | 16 name: "in_backprop"
74 name: "in_backprop"
134 name: "in_backprop"
195 name: "in_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | Dilation2DBackpropInput.pbtxt | 16 name: "in_backprop"
74 name: "in_backprop"
134 name: "in_backprop"
195 name: "in_backprop"
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 116 xla::StatusOr<xla::XlaOp> in_backprop = MakeXlaBackpropInputConvOp( in Compile() local
119 OP_REQUIRES_OK(ctx, in_backprop.status()); in Compile()
120 ctx->SetOutput(0, in_backprop.ValueOrDie()); in Compile()
|
| D | pooling_ops.cc | 455 auto in_backprop = in Compile() local
464 xla::ConvertElementType(in_backprop, xla_out_backprop_type)); in Compile()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_Dilation2DBackpropInput.pbtxt | 22 name: "in_backprop"
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 275 auto in_backprop = in AvgPoolGrad() local
288 return Pad(in_backprop, zero, remove_padding_config); in AvgPoolGrad()
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 21296 name: "in_backprop"
21354 name: "in_backprop"
21414 name: "in_backprop"
21475 name: "in_backprop"
|
| D | ops_history.v0.pbtxt | 10589 name: "in_backprop"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 19117 …, out_backprop tf.Output, strides []int64, rates []int64, padding string) (in_backprop tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 13246 name: "in_backprop"
|