Home
last modified time | relevance | path

Searched defs:conv_scale (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconv_autotuning.proto23 double conv_scale = 6; field
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dconv_parameters.h45 double conv_scale; member
H A Dconv_parameters.proto49 double conv_scale = 3; field
/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
H A Ddnn.cc161 dnn::DataType bias_type, dnn::DataType output_type, double conv_scale, in FusedConvolveRunnerFromDesc()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dgpu_utils.cc188 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults()
H A Dconv_ops_gpu.cc115 const se::dnn::ActivationMode activation_mode, double conv_scale, in AutotuneFusedConv()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc3613 bool SideInputNeeded(dnn::ActivationMode activation_mode, double conv_scale, in SideInputNeeded()
4882 double conv_scale, double side_input_scale, in Create()
5018 double conv_scale, double side_input_scale, in CudnnLegacyFusedConvRunner()
5065 dnn::DataType bias_type, dnn::DataType output_type, double conv_scale, in FusedConvolveRunnerFromDesc()
5140 dnn::DataType output_type, double conv_scale, double side_input_scale, in GetFusedConvolveRunners()
5757 DeviceMemoryBase conv_input_data, double conv_scale, in DoFusedConvolve()
/aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/
H A Detnaviv_ml_nn.c859 float conv_scale = (operation->input_scale * operation->weight_scale) / operation->output_scale; in create_nn_config() local