Home
last modified time | relevance | path

Searched refs:conv_input_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h282 const DeviceMemory<double>& conv_input_data, double conv_input_scale,
296 const DeviceMemory<float>& conv_input_data, float conv_input_scale,
310 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale,
326 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
340 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
618 const DeviceMemory<ElementType>& conv_input_data,
Dcuda_dnn.cc3179 const DeviceMemory<ElementType>& conv_input_data, in DoFusedConvolveImpl() argument
3246 << "\nconv_input_data.opaque() = " << conv_input_data.opaque() in DoFusedConvolveImpl()
3271 /*srcDesc=*/conv_input_nd.handle(), /*srcData=*/conv_input_data.opaque(), in DoFusedConvolveImpl()
3758 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
3770 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3780 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3792 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3802 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3815 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3825 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h320 const DeviceMemory<double>& conv_input_data, double conv_input_scale,
334 const DeviceMemory<float>& conv_input_data, float conv_input_scale,
348 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale,
364 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
587 const DeviceMemory<float>& conv_input_data,
784 const DeviceMemory<T>& conv_input_data,
Drocm_dnn.cc3677 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
3693 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3709 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3726 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
4620 const DeviceMemory<T>& conv_input_data, in DoFusedConvolutionBiasActivationImpl() argument
4679 fusion_plan.Execute(conv_input_nd.handle(), conv_input_data.opaque(), in DoFusedConvolutionBiasActivationImpl()
4708 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation() argument
4718 stream, miopenFloat, conv_input_descriptor, conv_input_data, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1145 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
1163 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
1182 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
1203 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
2489 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation() argument
Dstream.cc448 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in FusedConvolveWithAlgorithm() argument
459 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
468 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
479 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm() argument
490 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
499 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
510 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm() argument
522 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
531 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
542 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream.h356 const DeviceMemory<double> &conv_input_data, double conv_input_scale,
370 const DeviceMemory<float> &conv_input_data, float conv_input_scale,
384 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale,
399 const DeviceMemory<int8> &conv_input_data, float conv_input_scale,
413 const DeviceMemory<int8> &conv_input_data, float conv_input_scale,