Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h266 const DeviceMemory<double>& conv_input_data, double conv_input_scale,
280 const DeviceMemory<float>& conv_input_data, float conv_input_scale,
294 const DeviceMemory<Eigen::half>& conv_input_data,
312 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
535 const DeviceMemory<float>& conv_input_data,
709 const DeviceMemory<T>& conv_input_data,
Drocm_dnn.cc3147 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
3164 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3181 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3199 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
4024 const DeviceMemory<T>& conv_input_data, in DoFusedConvolutionBiasActivationImpl() argument
4083 fusion_plan.Execute(conv_input_nd.handle(), conv_input_data.opaque(), in DoFusedConvolutionBiasActivationImpl()
4112 const DeviceMemory<float>& conv_input_data, in DoFusedConvolutionBiasActivation() argument
4122 stream, miopenFloat, conv_input_descriptor, conv_input_data, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h275 const DeviceMemory<double>& conv_input_data, double conv_input_scale,
289 const DeviceMemory<float>& conv_input_data, float conv_input_scale,
303 const DeviceMemory<Eigen::half>& conv_input_data,
321 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
584 const DeviceMemory<ElementType>& conv_input_data,
Dcuda_dnn.cc2924 const DeviceMemory<ElementType>& conv_input_data, in DoFusedConvolveImpl() argument
2988 << "\nconv_input_data.opaque() = " << conv_input_data.opaque() in DoFusedConvolveImpl()
3015 /*srcDesc=*/conv_input_nd.handle(), /*srcData=*/conv_input_data.opaque(), in DoFusedConvolveImpl()
3345 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve() argument
3358 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3369 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3382 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3393 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
3407 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3418 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1117 const DeviceMemory<double>& conv_input_data, double conv_input_scale,
1134 const DeviceMemory<float>& conv_input_data, float conv_input_scale,
1152 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale,
1172 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
2354 const DeviceMemory<float>& conv_input_data,
Dstream.cc442 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in ThenFusedConvolveWithAlgorithm() argument
453 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm()
463 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm()
480 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm() argument
491 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm()
501 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm()
518 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm() argument
530 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm()
540 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm()
557 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dstream.h339 const DeviceMemory<double> &conv_input_data, double conv_input_scale,
353 const DeviceMemory<float> &conv_input_data, float conv_input_scale,
367 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale,
382 const DeviceMemory<int8> &conv_input_data, float conv_input_scale,