Home
last modified time | relevance | path

Searched refs:conv_input_scale (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op.py35 conv_input_scale=1.0, argument
103 conv_input_scale,
Dfused_conv2d_bias_activation_op_test_base.py670 def _SimulateFusedConv2dBiasActivationInt8(conv_input_scale, conv_input, kernel, argument
697 data_format="NCHW") * conv_input_scale
867 conv_input_scale = test_param["conv_input_scale"]
919 conv_input_scale=conv_input_scale,
927 conv_input_scale, conv_input, kernel, padding_type, strides,
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.h40 const Tensor& conv_input, ScaleType conv_input_scale,
54 const Tensor& conv_input, ScaleType conv_input_scale,
Dfused_conv2d_bias_activation_op.cc189 auto conv_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local
240 launcher_.launch(context, cudnn_use_autotune_, conv_input, conv_input_scale, in Compute()
442 const Tensor& conv_input_param, ScaleType conv_input_scale, in launch() argument
723 conv_input_desc, conv_input_ptr, conv_input_scale, in launch()
757 conv_input_desc, conv_input_ptr, conv_input_scale, filter_desc, in launch()
/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,
304 float conv_input_scale,
321 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
585 ScaleType conv_input_scale,
Dcuda_dnn.cc2925 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl() argument
2986 VLOG(2) << "\nconv_input_scale = " << conv_input_scale in DoFusedConvolveImpl()
3014 /*alpha1=*/&conv_input_scale, 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/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,
295 float conv_input_scale,
312 const DeviceMemory<int8>& conv_input_data, float conv_input_scale,
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
/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,
Dstream.cc442 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in ThenFusedConvolveWithAlgorithm() argument
454 PARAM(conv_input_scale), PARAM(filter_descriptor), 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
492 PARAM(conv_input_scale), PARAM(filter_descriptor), 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
531 PARAM(conv_input_scale), PARAM(filter_descriptor), 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,
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1178 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
1200 FusedChildContext(op_context, "Mul", output, {output, conv_input_scale}), in PredictFusedConv2DBiasActivation()