Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h279 const DeviceMemory<double>& side_input_data, double side_input_scale,
293 const DeviceMemory<float>& side_input_data, float side_input_scale,
308 const DeviceMemory<Eigen::half>& side_input_data,
325 const DeviceMemory<int8>& side_input_data, float side_input_scale,
589 const DeviceMemory<ElementType>& side_input_data,
Dcuda_dnn.cc2928 const DeviceMemory<ElementType>& side_input_data, in DoFusedConvolveImpl() argument
2984 : side_input_data.opaque(); in DoFusedConvolveImpl()
3349 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3360 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3373 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3384 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3397 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3409 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
3422 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3441 side_input_data, side_input_scale, bias_descriptor, biases, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h270 const DeviceMemory<double>& side_input_data, double side_input_scale,
284 const DeviceMemory<float>& side_input_data, float side_input_scale,
299 const DeviceMemory<Eigen::half>& side_input_data,
316 const DeviceMemory<int8>& side_input_data, float side_input_scale,
Drocm_dnn.cc3151 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3168 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3185 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3203 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1121 const DeviceMemory<double>& side_input_data, double side_input_scale,
1138 const DeviceMemory<float>& side_input_data, float side_input_scale,
1156 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale,
1176 const DeviceMemory<int8>& side_input_data, float side_input_scale,
Dstream.cc446 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm() argument
456 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm()
465 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
484 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument
494 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm()
503 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
522 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument
533 PARAM(side_input_data), PARAM(side_input_scale), in ThenFusedConvolveWithAlgorithm()
542 side_input_data, side_input_scale, bias_descriptor, biases, in ThenFusedConvolveWithAlgorithm()
561 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dstream.h343 const DeviceMemory<double> &side_input_data, double side_input_scale,
357 const DeviceMemory<float> &side_input_data, float side_input_scale,
371 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale,
386 const DeviceMemory<int8> &side_input_data, float side_input_scale,