Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1145 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
1163 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
1182 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
1203 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
Dstream.cc448 const DeviceMemory<double> &conv_input_data, double conv_input_scale, in FusedConvolveWithAlgorithm()
479 const DeviceMemory<float> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
510 const DeviceMemory<Eigen::half> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
542 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
573 const DeviceMemory<int8> &conv_input_data, float conv_input_scale, in FusedConvolveWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3180 ScaleType conv_input_scale, const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolveImpl()
3758 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3780 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3802 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3825 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3856 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3677 const DeviceMemory<double>& conv_input_data, double conv_input_scale, in DoFusedConvolve()
3693 const DeviceMemory<float>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3709 const DeviceMemory<Eigen::half>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
3726 const DeviceMemory<int8>& conv_input_data, float conv_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1746 auto& conv_input_scale = op_context.op_info.inputs(4); in PredictFusedConv2DBiasActivation() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3579 TRT_ShapedWeights conv_input_scale = inputs.at(4).weights(); in ConvertFusedConv2DBiasActivation() local