| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantize_and_dequantize_op_gpu.cu.cc | 62 typename TTypes<T>::Flat input_backprop, in operator ()()
66 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
78 typename TTypes<T, 3>::Tensor input_backprop, in operator ()()
82 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
|
| D | quantize_and_dequantize_op.cc | 154 Tensor* input_backprop = nullptr; in Compute() local
156 ctx->allocate_output(0, input.shape(), &input_backprop)); in Compute()
188 input_max_tensor.scalar<T>(), input_backprop->template flat<T>(), in Compute()
197 input_backprop->template flat_inner_outer_dims<T, 3>(axis_ - 1), in Compute()
378 typename TTypes<T>::Flat input_backprop, in operator ()()
382 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
394 typename TTypes<T, 3>::Tensor input_backprop, in operator ()()
398 d, gradient, input, input_min_tensor, input_max_tensor, input_backprop, in operator ()()
|
| D | quantize_and_dequantize_op.h | 69 typename TTypes<T>::Flat input_backprop,
80 typename TTypes<T, 3>::Tensor input_backprop,
280 typename TTypes<T>::Flat input_backprop, in Compute()
288 input_backprop.device(d) = gradient * in_range; in Compute()
301 typename TTypes<T, 3>::Tensor input_backprop, in Compute()
316 input_backprop.template chip<1>(i).device(d) = gradient_chip * in_range; in Compute()
|
| D | cudnn_pooling_gpu.cc | 139 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute() argument
177 transformed_input_backprop = *input_backprop; in Compute()
273 input_backprop->tensor<T, 5>()); in Compute()
|
| D | cudnn_pooling_gpu.h | 63 Tensor* input_backprop);
|
| D | pooling_ops_common.cc | 453 Tensor* input_backprop = nullptr; in Compute() local
455 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute()
500 transformed_input_backprop = *input_backprop; in Compute()
553 Tensor transformed_input_backprop = *input_backprop; in Compute()
753 input_backprop->tensor<T, 4>()); in Compute()
|
| D | cudnn_rnn_ops.cc | 858 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward() argument
895 auto input_backprop_data = AsDeviceMemory<T>(input_backprop); in DoBackward()
1674 Tensor input_backprop; in MaybeAutoTune() local
1687 data_type, model_shapes.input_shape, &input_backprop)); in MaybeAutoTune()
1738 &input_backprop, &input_h_backprop, &input_c_backprop, in MaybeAutoTune()
1896 Tensor* input_backprop = nullptr; in ComputeImpl() local
1902 &input_backprop, &input_h_backprop, in ComputeImpl()
1921 output_h_backprop, output_c_backprop, reserve_space, input_backprop, in ComputeImpl()
2003 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs() argument
2010 context->allocate_output(0, input_shape, input_backprop)); in AllocateOutputs()
|
| D | pooling_ops_3d.cc | 347 Tensor* input_backprop; in Compute() local
349 context->allocate_output(0, output_shape, &input_backprop)); in Compute()
366 padding, data_format_, input_backprop); in Compute()
773 TensorFormat data_format, Tensor* input_backprop) { in launch()
778 &tensor_out, input_backprop); in launch()
|
| D | maxpooling_op.cc | 64 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper() argument
66 if (input_backprop != nullptr) { in SpatialMaxPoolWithArgMaxHelper()
108 auto shard = [¶ms, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
177 if (input_backprop != nullptr) { in SpatialMaxPoolWithArgMaxHelper()
178 auto input_backprop_flat = input_backprop->flat<T>(); in SpatialMaxPoolWithArgMaxHelper()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | fractional_avg_pool_op_test.py | 378 input_backprop = self.evaluate(input_backprop_tensor)
388 self.assertShapeEqual(input_backprop, fap_input_backprop_tensor)
389 self.assertAllClose(input_backprop, fap_input_backprop)
417 input_backprop = self.evaluate(input_backprop_tensor)
429 self.assertShapeEqual(input_backprop, fap_input_backprop_tensor)
430 self.assertAllClose(input_backprop, fap_input_backprop)
|
| D | fractional_max_pool_op_test.py | 384 input_backprop = self.evaluate(input_backprop_tensor)
395 self.assertShapeEqual(input_backprop, fmp_input_backprop_tensor)
396 self.assertAllClose(input_backprop, fmp_input_backprop)
419 input_backprop = self.evaluate(input_backprop_tensor)
432 self.assertShapeEqual(input_backprop, fmp_input_backprop_tensor)
433 self.assertAllClose(input_backprop, fmp_input_backprop)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CudnnRNNBackpropV3.pbtxt | 56 name: "input_backprop"
202 name: "input_backprop"
355 name: "input_backprop"
|
| D | CudnnRNNBackpropV2.pbtxt | 52 name: "input_backprop"
|
| D | CudnnRNNBackprop.pbtxt | 48 name: "input_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | QuantizeAndDequantizeV4Grad.pbtxt | 20 name: "input_backprop"
|
| D | CudnnRNNBackpropV3.pbtxt | 56 name: "input_backprop"
202 name: "input_backprop"
|
| D | CudnnRNNBackpropV2.pbtxt | 52 name: "input_backprop"
|
| D | CudnnRNNBackprop.pbtxt | 48 name: "input_backprop"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CudnnRNNBackprop.pbtxt | 36 input_backprop: The backprop to input in the forward pass. Has the same shape
|
| D | api_def_CudnnRNNBackpropV2.pbtxt | 40 input_backprop: The backprop to input in the forward pass. Has the same shape
|
| D | api_def_CudnnRNNBackpropV3.pbtxt | 45 input_backprop: The backprop to input in the forward pass. Has the same shape
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 104 grad_outputs->push_back(qdq_v4_grad.input_backprop); in QuantizeAndDequantizeV4GradHelper()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 598 …, input_max tf.Output, optional ...QuantizeAndDequantizeV4GradAttr) (input_backprop tf.Output, inp…
5502 …utput, host_reserved tf.Output, optional ...CudnnRNNBackpropV3Attr) (input_backprop tf.Output, inp…
5613 ….Output, reserve_space tf.Output, optional ...CudnnRNNBackpropAttr) (input_backprop tf.Output, inp…
11278 …utput, host_reserved tf.Output, optional ...CudnnRNNBackpropV2Attr) (input_backprop tf.Output, inp…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 9680 name: "input_backprop"
9822 name: "input_backprop"
9968 name: "input_backprop"
30359 name: "input_backprop"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 16790 name: "input_backprop"
16932 name: "input_backprop"
17078 name: "input_backprop"
|