Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.cc116 double side_input_scale, double leakyrelu_alpha, in AutotuneFusedConv() argument
138 element_type, element_type, conv_scale, side_input_scale, in AutotuneFusedConv()
161 conv_desc, conv_scale, side_input_scale, activation_mode, in AutotuneFusedConv()
189 element_type, element_type, conv_scale, side_input_scale, in AutotuneFusedConv()
202 conv_desc, conv_scale, side_input_scale, activation_mode, in AutotuneFusedConv()
231 double side_input_scale, double leakyrelu_alpha,
247 double side_input_scale, double leakyrelu_alpha,
264 double side_input_scale, double leakyrelu_alpha,
H A Dconv_ops_gpu.h111 double side_input_scale, double leakyrelu_alpha,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_fused_conv_rewriter_test.cc643 EXPECT_EQ(config.side_input_scale(), 0.25); in TEST_F()
1083 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1124 EXPECT_EQ(config.side_input_scale(), 42); in TEST_F()
1163 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1245 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1293 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1346 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1398 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
1456 EXPECT_EQ(config.side_input_scale(), 1); in TEST_F()
H A Dgpu_conv_runner.cc127 if (params.config->fusion->side_input_scale != 0) { in RunGpuConvForwardActivation()
162 params.config->fusion->side_input_scale, in RunGpuConvForwardActivation()
333 fusion.side_input_scale = backend_config.side_input_scale(); in GetGpuConvConfig()
H A Dcudnn_fused_conv_rewriter.cc356 if (config.side_input_scale() != 1) { in FuseSideInputAlpha()
738 if (config->side_input_scale() != 0 && config->side_input_scale() != 1) { in VlogStats()
H A Dbackend_configs.proto40 double side_input_scale = 5; field
H A Dgpu_conv_runner.h52 double side_input_scale; member
H A Djitrt_custom_calls.cc707 double side_input_scale; member
796 side_input->side_input_scale); in GetConvDescriptor()
822 Optional<double> side_input_scale = llvm::None) const { in operator ()()
828 if (side_input_scale.has_value()) side_input_attrs = {*side_input_scale}; in operator ()()
H A Dgpu_conv_algorithm_picker.cc141 /* side_input_scale = */ config.fusion->side_input_scale, in GetAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dlazy_op_runner.h146 double conv_scale, side_input_scale, leakyrelu_alpha; member
160 config.output_type, config.conv_scale, config.side_input_scale, in RunnerFromAlgorithmDesc()
H A Dstream.h373 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm() argument
386 convolution_descriptor, side_input_data, side_input_scale, in FusedConvolveWithAlgorithm()
414 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc() argument
427 conv_input_scale, side_input_scale, leakyrelu_alpha, input_descriptor, in FusedConvolveRunnerFromDesc()
H A Dstream_executor_pimpl.cc317 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners() argument
332 conv_input_scale, side_input_scale, leakyrelu_alpha, stream, in GetFusedConvolveRunners()
H A Ddnn.cc145 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners() argument
162 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc() argument
H A Ddnn.h1308 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve() argument
1414 double side_input_scale, double leakyrelu_alpha, Stream* stream,
1428 double side_input_scale, double leakyrelu_alpha,
H A Dstream_executor_pimpl.h388 double side_input_scale, double leakyrelu_alpha, Stream* stream,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo_gpu/IR/
H A Dlhlo_gpu_ops.td107 // side_input * side_input_scale +
121 F64Attr:$side_input_scale)>.attributes);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
H A Dhlo_text_to_lhlo_no_opt.hlotxt313 …type\":\"DEFAULT_MATH\"},\"conv_result_scale\":1,\"activation_mode\":\"0\",\"side_input_scale\":0}"
340 …type\":\"DEFAULT_MATH\"},\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":0}"
359 // CHECK-SAME: side_input_scale = 1.000000e+00
369 …type\":\"DEFAULT_MATH\"},\"conv_result_scale\":1,\"activation_mode\":\"2\",\"side_input_scale\":1}"
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dconv_parameters.h46 double side_input_scale; member
H A Dconv_parameters.cc59 fusion_proto.set_side_input_scale(fusion_info.value().side_input_scale); in ConvParameters()
H A Dconv_parameters.proto50 double side_input_scale = 4; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h244 dnn::DataType output_type, double conv_scale, double side_input_scale,
260 double side_input_scale, double leakyrelu_alpha,
355 DeviceMemoryBase side_input_data, double side_input_scale,
H A Dcuda_dnn.cc3614 double side_input_scale) { in SideInputNeeded() argument
3623 bool check_scale = conv_scale != 1.0 || side_input_scale != 0.0; in SideInputNeeded()
4882 double conv_scale, double side_input_scale, in Create() argument
4903 workspace_size, input_type, conv_scale, side_input_scale, in Create()
5018 double conv_scale, double side_input_scale, in CudnnLegacyFusedConvRunner() argument
5032 side_input_scale_(side_input_scale), in CudnnLegacyFusedConvRunner()
5066 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc() argument
5104 conv_scale, side_input_scale, std::move(conv_input_nd), in FusedConvolveRunnerFromDesc()
5116 side_input_scale, leakyrelu_alpha, input_descriptor, in FusedConvolveRunnerFromDesc()
5124 SideInputNeeded(activation_mode, conv_scale, side_input_scale); in FusedConvolveRunnerFromDesc()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/lhlo_to_jitrt/
H A Dconv_to_jitrt.mlir280 // CHECK-DAG: side_input_scale = 1.000000e+00 : f64
305 side_input_scale = 1.000000e+00 : f64
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_graph_dumper.cc1219 if (config.side_input_scale() != 0 && config.side_input_scale() != 1) { in ExtractCudnnConvBackendConfigProps()
1220 props.emplace_back("side_input_scale", StrCat(config.side_input_scale())); in ExtractCudnnConvBackendConfigProps()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/lhlo_gpu/
H A Dlhlo_gpu_ops.mlir184 side_input_scale = 1.000000e+00 : f64

12