| /external/tensorflow/tensorflow/core/kernels/ |
| cudnn_pooling_gpu.h | 63 Tensor* input_backprop);
|
| cudnn_pooling_gpu.cc | 124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) {
162 transformed_input_backprop = *input_backprop;
242 input_backprop->tensor<T, 5>());
|
| pooling_ops_common.cc | 250 Tensor* input_backprop = nullptr; local
252 context->allocate_output(0, tensor_in_shape, &input_backprop));
294 transformed_input_backprop = *input_backprop;
385 input_backprop->tensor<T, 4>());
|
| pooling_ops_3d_sycl.h | 254 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_); local
286 const T* input_backprop_n = input_backprop + n * p_.out_planes_ *
425 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_); local
460 output_backprop[index] = input_backprop[n * p_.in_planes_ * p_.in_rows_ *
654 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_); local
680 const T* input_backprop_n = input_backprop + n * p_.out_planes_ *
[all...] |
| pooling_ops_3d.cc | 351 Tensor* input_backprop; variable
353 context->allocate_output(0, output_shape, &input_backprop));
370 padding, data_format_, input_backprop);
[all...] |
| maxpooling_op.cc | 58 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop,
90 auto shard = [¶ms, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop,
155 auto input_backprop_flat = input_backprop->flat<T>();
[all...] |
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fractional_avg_pool_op_test.py | 410 input_backprop = input_backprop_tensor.eval()
420 self.assertShapeEqual(input_backprop, fap_input_backprop_tensor)
421 self.assertAllClose(input_backprop, fap_input_backprop)
449 input_backprop = input_backprop_tensor.eval()
461 self.assertShapeEqual(input_backprop, fap_input_backprop_tensor)
462 self.assertAllClose(input_backprop, fap_input_backprop)
|
| fractional_max_pool_op_test.py | 393 input_backprop = input_backprop_tensor.eval()
404 self.assertShapeEqual(input_backprop, fmp_input_backprop_tensor)
405 self.assertAllClose(input_backprop, fmp_input_backprop)
431 input_backprop = input_backprop_tensor.eval()
444 self.assertShapeEqual(input_backprop, fmp_input_backprop_tensor)
445 self.assertAllClose(input_backprop, fmp_input_backprop)
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| cudnn_rnn_ops.cc | 1017 Tensor* input_backprop = nullptr; variable
[all...] |
