Searched defs:conv_scale (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | conv_autotuning.proto | 23 double conv_scale = 6; field
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | conv_parameters.h | 45 double conv_scale; member
|
| D | conv_parameters.proto | 49 double conv_scale = 3; field
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | lazy_op_runner.h | 146 double conv_scale, side_input_scale, leakyrelu_alpha; member
|
| D | dnn.cc | 161 dnn::DataType bias_type, dnn::DataType output_type, double conv_scale, in FusedConvolveRunnerFromDesc()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.cc | 188 const se::dnn::ConvolutionDescriptor& conv_desc, double conv_scale, in LogFusedConvForwardAutotuneResults()
|
| D | conv_ops_gpu.cc | 115 const se::dnn::ActivationMode activation_mode, double conv_scale, in AutotuneFusedConv()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3613 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()
|