Home
last modified time | relevance | path

Searched defs:input_backprop_data (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_pooling_gpu.cc231 auto input_backprop_data = in Compute() local
H A Dpooling_ops_common.cc709 auto input_backprop_data = AsDeviceMemory( in Compute() local
H A Dconv_grad_ops_3d.cc528 T* input_backprop_data = in_backprop_flat.data(); in Compute() local
H A Dcudnn_rnn_ops.cc918 auto input_backprop_data = AsDeviceMemory<T>(input_backprop); in DoBackward() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2238 DeviceMemory<Eigen::half>* input_backprop_data, in DoRnnBackward()
2267 DeviceMemory<float>* input_backprop_data, in DoRnnBackward()
2296 DeviceMemory<double>* input_backprop_data, in DoRnnBackward()
H A Dstream.cc2156 DeviceMemory<Eigen::half> *input_backprop_data, in ThenRnnBackward()
2201 DeviceMemory<float> *input_backprop_data, in ThenRnnBackward()
2247 DeviceMemory<double> *input_backprop_data, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2268 DeviceMemory<T>* input_backprop_data, in DoRnnBackwardImpl()
2760 DeviceMemory<Eigen::half>* input_backprop_data, in DoRnnBackward()
2812 DeviceMemory<float>* input_backprop_data, in DoRnnBackward()
2865 DeviceMemory<double>* input_backprop_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc2160 DeviceMemory<T>* input_backprop_data, in DoRnnBackwardImpl()
2619 DeviceMemory<Eigen::half>* input_backprop_data, in DoRnnBackward()
2671 DeviceMemory<float>* input_backprop_data, in DoRnnBackward()
2724 DeviceMemory<double>* input_backprop_data, in DoRnnBackward()