| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 126 if (params.config.fusion->side_input_scale != 0) { in RunGpuConvForwardActivation()
144 params.config.fusion->side_input_scale, bias_desc, in RunGpuConvForwardActivation()
344 fusion.side_input_scale = backend_config.side_input_scale(); in GetGpuConvConfig()
|
| D | gpu_conv_runner.h | 61 double side_input_scale; member
|
| D | backend_configs.proto | 42 double side_input_scale = 5; field
|
| D | ir_emitter_unnested.cc | 1015 conv.side_input_scale().convertToDouble()); in EmitConvolutionThunk()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | lhlo_gpu_ops.td | 175 // side_input * side_input_scale +
189 F64Attr:$side_input_scale)>.attributes);
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 224 double conv_input_scale, double side_input_scale, Stream* stream,
328 DeviceMemoryBase side_input_data, double side_input_scale,
343 DeviceMemoryBase side_input_data, double side_input_scale,
|
| D | cuda_dnn.cc | 4302 double conv_input_scale, double side_input_scale, Stream* stream, in GetFusedConvolveExecutionPlans() argument
4313 kind, element_type, conv_input_scale, side_input_scale, stream, in GetFusedConvolveExecutionPlans()
4875 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
4948 (side_input_scale == 0) ? output_data.opaque() : side_input_data.opaque(); in DoFusedConvolve()
4960 << "\nside_input_scale = " << side_input_scale in DoFusedConvolve()
4985 /*alpha2=*/ScalingParam(side_input_scale).ToVoidPointer(input_type), in DoFusedConvolve()
5022 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolveWithExecutionPlan() argument
5042 side_input_scale, stream, conv_input_descriptor, filter_descriptor, in DoFusedConvolveWithExecutionPlan()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/ |
| D | hlo_text_to_lhlo_no_opt.hlotxt | 360 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"0\",\"side_input_scale\":0}"
387 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":0}"
406 // CHECK-SAME: side_input_scale = 1.000000e+00
416 …nsor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":1}"
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 381 double conv_input_scale, double side_input_scale, Stream *stream, in GetFusedConvolveExecutionPlans() argument
396 kind, element_type, conv_input_scale, side_input_scale, stream, in GetFusedConvolveExecutionPlans()
|
| D | stream.h | 393 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm() argument
406 convolution_descriptor, side_input_data, side_input_scale, in FusedConvolveWithAlgorithm()
421 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithExecutionPlan() argument
435 convolution_descriptor, side_input_data, side_input_scale, in FusedConvolveWithExecutionPlan()
|
| D | dnn.h | 1260 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 329 double side_input_scale, const ConvLaunch launch, OpKernelContext* context,
347 side_input_scale, stream, input_desc, filter_desc, bias_desc,
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | lhlo_gpu_ops.mlir | 174 side_input_scale = 1.000000e+00 : f64}
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 333 DeviceMemoryBase side_input_data, double side_input_scale,
|
| D | rocm_dnn.cc | 3678 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 1759 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
1785 {side_input, side_input_scale})); in PredictFusedConv2DBiasActivation()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 952 builder_.getF64FloatAttr(backend_config.side_input_scale())); in EmitDnnConvolution()
|