Home
last modified time | relevance | path

Searched refs:conv_input_scale (Results 1 – 9 of 9) 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,
619 ScaleType conv_input_scale,
Dcuda_dnn.cc3180 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl() argument
3244 VLOG(2) << "\nconv_input_scale = " << conv_input_scale in DoFusedConvolveImpl()
3270 /*alpha1=*/&conv_input_scale, 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,
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
/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
Dstream.cc448 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in FusedConvolveWithAlgorithm() argument
460 PARAM(conv_input_scale), PARAM(filter_descriptor), 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
491 PARAM(conv_input_scale), PARAM(filter_descriptor), 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
523 PARAM(conv_input_scale), PARAM(filter_descriptor), 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,
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1746 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
1764 FusedChildContext(op_context, "Mul", output, {output, conv_input_scale}), in PredictFusedConv2DBiasActivation()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3579 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local
3580 if (conv_input_scale.count() != 1 || in ConvertFusedConv2DBiasActivation()
3581 conv_input_scale.TrtDType() != nvinfer1::DataType::kFLOAT || in ConvertFusedConv2DBiasActivation()
3582 conv_input_scale.GetSpan<float>()[0] != 1.0) { in ConvertFusedConv2DBiasActivation()