Home
last modified time | relevance | path

Searched defs:side_input_scale (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto42 double side_input_scale = 5; field
Dgpu_conv_runner.h61 double side_input_scale; member
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1149 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
1167 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
1186 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
1207 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
Dstream.cc452 const DeviceMemory<double> &side_input_data, double side_input_scale, in FusedConvolveWithAlgorithm()
483 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
514 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
546 const DeviceMemory<int8> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
577 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3183 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
3762 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3784 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3806 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3829 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
3860 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3681 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3697 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3713 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3730 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1747 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local