Home
last modified time | relevance | path

Searched refs:side_input_scale (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc114 if (params.config.fusion->side_input_scale != 0) { in RunGpuConvForwardActivation()
132 params.config.fusion->side_input_scale, bias_desc, in RunGpuConvForwardActivation()
310 fusion.side_input_scale = backend_config.side_input_scale(); in GetGpuConvConfig()
Dgpu_conv_runner.h61 double side_input_scale; member
Dbackend_configs.proto42 double side_input_scale = 5; field
Dir_emitter_unnested.cc1212 op.side_input_scale().convertToDouble()); in EmitConvolutionThunkFromMlir()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h286 const DeviceMemory<double>& side_input_data, double side_input_scale,
300 const DeviceMemory<float>& side_input_data, float side_input_scale,
314 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale,
330 const DeviceMemory<int8>& side_input_data, float side_input_scale,
344 const DeviceMemory<float>& side_input_data, float side_input_scale,
624 ScaleType side_input_scale, const dnn::BatchDescriptor& bias_descriptor,
Dcuda_dnn.cc3183 const DeviceMemory<OutputType>& side_input_data, ScaleType side_input_scale, in DoFusedConvolveImpl() argument
3241 auto side_input_data_ptr = (side_input_scale == 0) ? output_data->opaque() in DoFusedConvolveImpl()
3253 << "\nside_input_scale = " << side_input_scale in DoFusedConvolveImpl()
3275 /*workSpaceSizeInBytes=*/scratch.size(), /*alpha2=*/&side_input_scale, in DoFusedConvolveImpl()
3762 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3772 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
3784 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3794 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
3806 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3817 side_input_scale, bias_descriptor, biases, activation_mode, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.td147 // side_input * side_input_scale +
160 F64Attr:$side_input_scale)>.attributes);
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt307 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"0\",\"side_input_scale\":0}"
346 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":0}"
375 // CHECK-SAME: side_input_scale = 1.000000e+00
386 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":1}"
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h324 const DeviceMemory<double>& side_input_data, double side_input_scale,
338 const DeviceMemory<float>& side_input_data, float side_input_scale,
352 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale,
368 const DeviceMemory<int8>& side_input_data, float side_input_scale,
Drocm_dnn.cc3681 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
3697 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3713 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
3730 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1149 const DeviceMemory<double>& side_input_data, double side_input_scale, in DoFusedConvolve() argument
1167 const DeviceMemory<float>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
1186 const DeviceMemory<Eigen::half>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
1207 const DeviceMemory<int8>& side_input_data, float side_input_scale, in DoFusedConvolve() argument
Dstream.cc452 const DeviceMemory<double> &side_input_data, double side_input_scale, in FusedConvolveWithAlgorithm() argument
462 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
470 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
483 const DeviceMemory<float> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
493 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
501 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
514 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
525 PARAM(side_input_data), PARAM(side_input_scale), in FusedConvolveWithAlgorithm()
533 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
546 const DeviceMemory<int8> &side_input_data, float side_input_scale, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream.h360 const DeviceMemory<double> &side_input_data, double side_input_scale,
374 const DeviceMemory<float> &side_input_data, float side_input_scale,
388 const DeviceMemory<Eigen::half> &side_input_data, float side_input_scale,
403 const DeviceMemory<int8> &side_input_data, float side_input_scale,
417 const DeviceMemory<float> &side_input_data, float side_input_scale,
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir180 side_input_scale = 1.000000e+00 : f64,
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1747 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
1773 {side_input, side_input_scale})); in PredictFusedConv2DBiasActivation()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc838 builder_.getF64FloatAttr(backend_config.side_input_scale())); in EmitDnnConvolution()