| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.cc | 114 Tensor* in_backprop, TensorFormat data_format) { in operator ()()
117 d, in_backprop->tensor<T, 4>(), filter.tensor<T, 4>(), in operator ()()
333 Tensor* in_backprop = nullptr; in Compute() local
335 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
364 in_backprop->tensor<T, 4>(), filter.tensor<T, 4>(), in Compute()
459 auto in_backprop_flat = in_backprop->template flat<T>(); in Compute()
645 Tensor* in_backprop = nullptr; in Compute() local
647 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
663 explicit_paddings_, in_backprop, data_format_); in Compute()
684 const std::vector<int64>& explicit_paddings, Tensor* in_backprop, in operator ()() argument
[all …]
|
| 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 | 238 Tensor* in_backprop; in Compute() local
240 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
244 in_backprop->tensor<T, 5>(), // input_backward in Compute()
344 Tensor* in_backprop; in Compute() local
346 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
434 in_backprop->tensor<T, 5>(), // input_backward in Compute()
463 auto in_backprop_flat = in_backprop->template flat<T>(); in Compute()
1160 Tensor* in_backprop; in Compute() local
1162 context->allocate_output(0, input_shape, &in_backprop)); in Compute()
1181 auto c_ptr = AsDeviceMemory(in_backprop->template flat<T>().data(), in Compute()
[all …]
|
| D | depthwise_conv_grad_op.cc | 376 T* in_backprop, TensorFormat data_format) { in operator ()()
405 auto shard = [&ctx, &args, &out_backprop, &filter_data, &in_backprop]( in operator ()()
446 in_backprop + b * input_image_size); in operator ()()
463 T* in_backprop) { in DepthwiseConvBackpropInputReference() argument
508 in_backprop[in_backprop_offset] = sum; in DepthwiseConvBackpropInputReference()
593 Tensor* in_backprop = nullptr; in Compute() local
595 {0}, 0, input_shape, &in_backprop)); in Compute()
637 in_backprop, data_format_); in Compute()
643 auto in_backprop_ptr = in_backprop->template flat<T>().data(); in Compute()
|
| 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_op_gpu.h | 818 const T* filter, T* in_backprop,
879 in_backprop[in_backprop_offset] = sum;
888 const T* filter, T* in_backprop,
956 in_backprop[in_backprop_offset] = sum;
965 const T* filter, T* in_backprop,
988 args, out_backprop, filter, in_backprop, num_in_backprop));
996 const T* filter, T* in_backprop,
1002 ctx, args, out_backprop, filter, in_backprop, data_format);
1007 ctx, args, out_backprop, filter, in_backprop, data_format);
1011 ctx, args, out_backprop, filter, in_backprop, data_format);
[all …]
|
| 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,
|
| D | dilation_ops.h | 46 typename TTypes<T, 4>::Tensor in_backprop);
|
| D | conv_grad_ops.h | 181 Tensor* in_backprop, TensorFormat data_format);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | conv_ops.cc | 106 xla::StatusOr<xla::XlaOp> in_backprop = in Compile() local
109 OP_REQUIRES_OK(ctx, in_backprop.status()); in Compile()
110 ctx->SetOutput(0, in_backprop.ValueOrDie()); in Compile()
|
| D | pooling_ops.cc | 434 auto in_backprop = in Compile() local
443 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 | 272 auto in_backprop = in AvgPoolGrad() local
285 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.v1.pbtxt | 21639 name: "in_backprop"
21697 name: "in_backprop"
21757 name: "in_backprop"
21818 name: "in_backprop"
|
| D | ops_history.v0.pbtxt | 10589 name: "in_backprop"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20891 …, out_backprop tf.Output, strides []int64, rates []int64, padding string) (in_backprop tf.Output) {
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9922 name: "in_backprop"
|