Home
last modified time | relevance | path

Searched defs:input_backprop (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_pooling_gpu.cc124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
Dpooling_ops_common.cc296 Tensor* input_backprop = nullptr; in Compute() local
Dcudnn_rnn_ops.cc799 Tensor* input_backprop, Tensor* input_h_backprop, Tensor* input_c_backprop, in DoBackward()
1488 Tensor input_backprop; in MaybeAutoTune() local
1697 Tensor* input_backprop = nullptr; in ComputeImpl() local
1803 Tensor** input_backprop, Tensor** input_h_backprop, in AllocateOutputs()
Dpooling_ops_3d.cc351 Tensor* input_backprop; in Compute() local
Dmaxpooling_op.cc60 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()