Home
last modified time | relevance | path

Searched refs:input_backprop (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dfractional_avg_pool_op_test.py378 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)
Dfractional_max_pool_op_test.py384 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/kernels/
Dpooling_ops_common.cc296 Tensor* input_backprop = nullptr; in Compute() local
298 context->allocate_output(0, tensor_in_shape, &input_backprop)); in Compute()
340 transformed_input_backprop = *input_backprop; in Compute()
431 input_backprop->tensor<T, 4>()); in Compute()
Dcudnn_pooling_gpu.cc124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute() argument
162 transformed_input_backprop = *input_backprop; in Compute()
242 input_backprop->tensor<T, 5>()); in Compute()
Dcudnn_pooling_gpu.h63 Tensor* input_backprop);
Dcudnn_rnn_ops.cc799 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward() argument
835 auto input_backprop_data = AsDeviceMemory<T>(input_backprop); in DoBackward()
1488 Tensor input_backprop; in MaybeAutoTune() local
1501 data_type, model_shapes.input_shape, &input_backprop)); in MaybeAutoTune()
1551 &input_backprop, &input_h_backprop, &input_c_backprop, in MaybeAutoTune()
1697 Tensor* input_backprop = nullptr; in ComputeImpl() local
1703 &input_backprop, &input_h_backprop, in ComputeImpl()
1722 output_h_backprop, output_c_backprop, reserve_space, input_backprop, in ComputeImpl()
1803 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs() argument
1809 context->allocate_output(0, input_shape, input_backprop)); in AllocateOutputs()
Dpooling_ops_3d_sycl.h254 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_);
286 const T* input_backprop_n = input_backprop + n * p_.out_planes_ *
425 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_);
460 output_backprop[index] = input_backprop[n * p_.in_planes_ * p_.in_rows_ *
654 T* input_backprop = ConvertToActualTypeSycl(T, input_backprop_accessor_);
680 const T* input_backprop_n = input_backprop + n * p_.out_planes_ *
Dpooling_ops_3d.cc351 Tensor* input_backprop; in Compute() local
353 context->allocate_output(0, output_shape, &input_backprop)); in Compute()
370 padding, data_format_, input_backprop); in Compute()
777 TensorFormat data_format, Tensor* input_backprop) { in launch()
782 &tensor_out, input_backprop); in launch()
Dmaxpooling_op.cc60 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper() argument
62 if (input_backprop != nullptr) { in SpatialMaxPoolWithArgMaxHelper()
100 auto shard = [&params, &in_mat, &out_mat, &out_arg_max_mat, &input_backprop, in SpatialMaxPoolWithArgMaxHelper()
168 if (input_backprop != nullptr) { in SpatialMaxPoolWithArgMaxHelper()
169 auto input_backprop_flat = input_backprop->flat<T>(); in SpatialMaxPoolWithArgMaxHelper()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CudnnRNNBackprop.pbtxt36 input_backprop: The backprop to input in the forward pass. Has the same shape
Dapi_def_CudnnRNNBackpropV2.pbtxt40 input_backprop: The backprop to input in the forward pass. Has the same shape
Dapi_def_CudnnRNNBackpropV3.pbtxt45 input_backprop: The backprop to input in the forward pass. Has the same shape
/external/tensorflow/tensorflow/go/op/
Dwrappers.go4798 …utput, host_reserved tf.Output, optional ...CudnnRNNBackpropV3Attr) (input_backprop tf.Output, inp…
4912 …utput, host_reserved tf.Output, optional ...CudnnRNNBackpropV2Attr) (input_backprop tf.Output, inp…
21878 ….Output, reserve_space tf.Output, optional ...CudnnRNNBackpropAttr) (input_backprop tf.Output, inp…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt7386 name: "input_backprop"
7528 name: "input_backprop"
7674 name: "input_backprop"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt16848 name: "input_backprop"
16990 name: "input_backprop"
17136 name: "input_backprop"
17282 name: "input_backprop"
Dops_history.v2.pbtxt16790 name: "input_backprop"
16932 name: "input_backprop"
17078 name: "input_backprop"