Home
last modified time | relevance | path

Searched refs:backward_input_data (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.cc828 DeviceMemory<double> *backward_input_data, in ThenConvolveBackwardDataWithAlgorithm() argument
835 PARAM(backward_input_data)); in ThenConvolveBackwardDataWithAlgorithm()
844 *backward_input_data, filter_descriptor, filter_data, in ThenConvolveBackwardDataWithAlgorithm()
853 backward_input_data, algorithm_desc, &scratch_memory, in ThenConvolveBackwardDataWithAlgorithm()
873 DeviceMemory<float> *backward_input_data, in ThenConvolveBackwardDataWithAlgorithm() argument
880 PARAM(backward_input_data)); in ThenConvolveBackwardDataWithAlgorithm()
889 *backward_input_data, filter_descriptor, filter_data, in ThenConvolveBackwardDataWithAlgorithm()
898 backward_input_data, algorithm_desc, &scratch_memory, in ThenConvolveBackwardDataWithAlgorithm()
918 DeviceMemory<Eigen::half> *backward_input_data, in ThenConvolveBackwardDataWithAlgorithm() argument
925 PARAM(backward_input_data)); in ThenConvolveBackwardDataWithAlgorithm()
[all …]
Ddnn.h1348 DeviceMemory<ElementType>* backward_input_data,
1355 *backward_input_data, filter_descriptor, filter_data,
Dstream.h411 DeviceMemory<double> *backward_input_data,
423 DeviceMemory<float> *backward_input_data,
435 DeviceMemory<Eigen::half> *backward_input_data,