Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto42 double side_input_scale = 5; field
Dgpu_conv_runner.h56 double side_input_scale; member
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3219 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl()
3784 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3808 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3832 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3857 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
3890 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3440 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve()
3457 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve()
3474 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve()
3492 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1318 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc466 const DeviceMemory<double> &side_input_data, double side_input_scale, in ThenFusedConvolveWithAlgorithm()
504 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
542 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
581 const DeviceMemory<int8> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()
619 const DeviceMemory<float> &side_input_data, float side_input_scale, in ThenFusedConvolveWithAlgorithm()