Searched defs:side_input_scale (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/ |
| H A D | conv_parameters.h | 46 double side_input_scale; member
|
| H A D | conv_parameters.proto | 50 double side_input_scale = 4; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | backend_configs.proto | 40 double side_input_scale = 5; field
|
| H A D | gpu_conv_runner.h | 52 double side_input_scale; member
|
| H A D | jitrt_custom_calls.cc | 707 double side_input_scale; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.h | 373 const DeviceMemory<SideInputT> &side_input_data, ScaleT side_input_scale, in FusedConvolveWithAlgorithm()
414 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
|
| H A D | lazy_op_runner.h | 146 double conv_scale, side_input_scale, leakyrelu_alpha; member
|
| H A D | dnn.cc | 145 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners()
162 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
|
| H A D | stream_executor_pimpl.cc | 317 dnn::DataType output_type, double conv_input_scale, double side_input_scale, in GetFusedConvolveRunners()
|
| H A D | dnn.h | 1308 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_gpu.cc | 116 double side_input_scale, double leakyrelu_alpha, in AutotuneFusedConv()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 3614 double side_input_scale) { in SideInputNeeded()
4882 double conv_scale, double side_input_scale, in Create()
5018 double conv_scale, double side_input_scale, in CudnnLegacyFusedConvRunner()
5066 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
5140 dnn::DataType output_type, double conv_scale, double side_input_scale, in GetFusedConvolveRunners()
5761 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | op_level_cost_estimator.cc | 1780 auto& side_input_scale = op_context.op_info.inputs(5); in PredictFusedConv2DBiasActivation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3803 DeviceMemoryBase side_input_data, double side_input_scale, in DoFusedConvolve()
|