Searched defs:input_backprop (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_pooling_gpu.cc | 124 const Tensor* tensor_in, const Tensor* tensor_out, Tensor* input_backprop) { in Compute()
|
D | pooling_ops_common.cc | 296 Tensor* input_backprop = nullptr; in Compute() local
|
D | cudnn_rnn_ops.cc | 799 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()
|
D | pooling_ops_3d.cc | 351 Tensor* input_backprop; in Compute() local
|
D | maxpooling_op.cc | 60 Tensor* input_backprop, const Tensor& tensor_in, const Tensor& out_backprop, in SpatialMaxPoolWithArgMaxHelper()
|