Home
last modified time | relevance | path

Searched defs:activation_mode (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1170 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
1190 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
1223 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
1241 dnn::ActivationMode activation_mode, in DoBatchNormalizationBackward()
1310 dnn::ActivationMode activation_mode, in DoFusedConvolve()
1688 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, in DoActivate()
2428 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()
2469 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, in DoFusedBatchNormActivationInference()
2482 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data, in DoFusedBatchNormActivationInference()
2521 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, in DoFusedBatchNormActivationForward()
[all …]
H A Dstream.cc336 dnn::ActivationMode activation_mode, DeviceMemory<float> *y, in ThenBatchNormalizationForward()
361 dnn::ActivationMode activation_mode, DeviceMemory<float> *x_backprop, in ThenBatchNormalizationBackward()
390 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y, in ThenBatchNormalizationForward()
416 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *x_backprop, in ThenBatchNormalizationBackward()
647 Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode, in ThenActivate()
655 Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode, in ThenActivateWithOptions()
H A Dstream.h375 const DeviceMemory<BiasT> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
420 dnn::ActivationMode activation_mode) { in FusedConvolveRunnerFromDesc()
H A Dlazy_op_runner.h152 ActivationMode activation_mode; member
H A Ddnn.cc152 dnn::ActivationMode activation_mode, in GetFusedConvolveRunners()
168 dnn::ActivationMode activation_mode) { in FusedConvolveRunnerFromDesc()
H A Dstream_executor_pimpl.cc324 dnn::ActivationMode activation_mode, in GetFusedConvolveRunners()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc992 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor()
3645 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
3665 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
3687 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl()
3733 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward()
3751 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
3805 dnn::ActivationMode activation_mode, in DoFusedConvolve()
4009 dnn::ActivationMode activation_mode, in DoActivate()
4564 const DeviceMemory<T>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivationImpl()
4652 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_ex_op_test.cc141 bool has_side_input, const string& activation_mode, in RunFusedBatchNorm()
229 bool has_side_input, const string& activation_mode, in RunFusedBatchNormEx()
451 const string& activation_mode) { in VerifyFusedBatchNormEx()
593 string activation_mode = side_input_and_activation.second; in FusedBatchNormEx() local
H A Dfused_batch_norm_op.cc57 string ToString(FusedBatchNormActivationMode activation_mode) { in ToString()
67 FusedBatchNormActivationMode* activation_mode) { in ParseActivationMode()
633 const FusedBatchNormActivationMode activation_mode) { in AsDnnActivationMode()
H A Dgpu_utils.cc189 double side_value_scale, se::dnn::ActivationMode activation_mode, 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.cc1016 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor()
3613 bool SideInputNeeded(dnn::ActivationMode activation_mode, double conv_scale, in SideInputNeeded()
3637 const dnn::ActivationMode activation_mode, CudnnHandle& cudnn) { in GetCudnnFusedOperationGraph()
5072 dnn::ActivationMode activation_mode) { in FusedConvolveRunnerFromDesc()
5147 const dnn::ActivationMode activation_mode, in GetFusedConvolveRunners()
5416 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
5440 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
5465 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl()
5611 dnn::ActivationMode activation_mode, DeviceMemory<float>* x_backprop, in DoBatchNormalizationBackward()
5632 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* x_backprop, in DoBatchNormalizationBackward()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dconv_parameters.h48 stream_executor::dnn::ActivationMode activation_mode; member
H A Dconv_parameters.proto48 stream_executor.dnn.ActivationMode activation_mode = 2; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbackend_configs.proto36 int64 activation_mode = 3; field
H A Dgpu_conv_algorithm_picker.cc465 auto activation_mode = in AutotuneOneConvRunner() local
H A Djitrt_custom_calls.cc703 se::dnn::ActivationMode activation_mode; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc59 string activation_mode; in FusedBatchNormOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op.cc53 FusedBNActivationMode activation_mode; member
/aosp_15_r20/tools/test/connectivity/acts/framework/acts/controllers/rohdeschwarz_lib/
Dcmw500.py522 def scc_activation_mode(self, activation_mode): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc1707 string activation_mode; in FusedBatchNormExRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc3093 string data_format, filter_format, activation_mode, padding_type; in ConvertFusedConv2DBiasActivation() local