Searched defs:conv_input_scale (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op.py | 35 conv_input_scale=1.0, argument
|
| D | fused_conv2d_bias_activation_op_test_base.py | 670 def _SimulateFusedConv2dBiasActivationInt8(conv_input_scale, conv_input, kernel, argument
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 189 auto conv_input_scale = *reinterpret_cast<const ScaleType*>( in Compute() local
442 const Tensor& conv_input_param, ScaleType conv_input_scale, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2925 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3345 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3369 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3393 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3418 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3147 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3164 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3181 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3199 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1178 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 442 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in ThenFusedConvolveWithAlgorithm()
480 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
518 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
557 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in ThenFusedConvolveWithAlgorithm()
|