Home
last modified time | relevance | path

Searched refs:backward_output_data (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.cc825 DeviceMemory<double> backward_output_data, in ThenConvolveBackwardDataWithAlgorithm() argument
833 PARAM(output_descriptor), PARAM(backward_output_data), in ThenConvolveBackwardDataWithAlgorithm()
845 output_descriptor, backward_output_data, in ThenConvolveBackwardDataWithAlgorithm()
852 backward_output_data, convolution_descriptor, input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
870 DeviceMemory<float> backward_output_data, in ThenConvolveBackwardDataWithAlgorithm() argument
878 PARAM(output_descriptor), PARAM(backward_output_data), in ThenConvolveBackwardDataWithAlgorithm()
890 output_descriptor, backward_output_data, in ThenConvolveBackwardDataWithAlgorithm()
897 backward_output_data, convolution_descriptor, input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
915 DeviceMemory<Eigen::half> backward_output_data, in ThenConvolveBackwardDataWithAlgorithm() argument
923 PARAM(output_descriptor), PARAM(backward_output_data), in ThenConvolveBackwardDataWithAlgorithm()
[all …]
Ddnn.h1345 const DeviceMemory<ElementType>& backward_output_data,
1356 output_descriptor, backward_output_data,
1392 const DeviceMemory<ElementType>& backward_output_data,
1403 output_descriptor, backward_output_data,
Dstream.h408 DeviceMemory<double> backward_output_data,
420 DeviceMemory<float> backward_output_data,
432 DeviceMemory<Eigen::half> backward_output_data,
444 DeviceMemory<double> backward_output_data,
456 DeviceMemory<float> backward_output_data,
468 DeviceMemory<Eigen::half> backward_output_data,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2775 BatchDescriptor* output_descriptor, DeviceMemoryBase backward_output_data, in MaybeTransformLayout() argument
2778 return backward_output_data; in MaybeTransformLayout()
2782 stream->AllocateTemporaryArray<uint8>(backward_output_data.size()) in MaybeTransformLayout()
2798 backward_output_data.opaque(), &alpha2, orig_out_back_nd.handle(), in MaybeTransformLayout()
2799 backward_output_data.opaque(), &beta, transformed_out_back_nd.handle(), in MaybeTransformLayout()