Home
last modified time | relevance | path

Searched defs:conv_input_scale (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op.py35 conv_input_scale=1.0, argument
Dfused_conv2d_bias_activation_op_test.py623 def SimulateFusedConv2dBiasActivationInt8(conv_input_scale, conv_input, kernel, argument
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc179 auto conv_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local
307 const Tensor& conv_input_param, ScaleType conv_input_scale, in launch()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2504 const DeviceMemory<Type>& conv_input_data, ScaleType conv_input_scale, in DoFusedConvolveImpl()
2970 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
2992 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3014 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3037 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc417 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
450 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
484 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithScratch()
571 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
609 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
648 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
748 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolve()