Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/autotune_maps/
Dconv_parameters.h48 stream_executor::dnn::ActivationMode activation_mode; member
Dconv_parameters.proto48 stream_executor.dnn.ActivationMode activation_mode = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto36 int64 activation_mode = 3; field
Dgpu_conv_algorithm_picker.cc465 auto activation_mode = in AutotuneOneConvRunner() local
Djitrt_custom_calls.cc703 se::dnn::ActivationMode activation_mode; member
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
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 …]
Dlazy_op_runner.h152 ActivationMode activation_mode; member
Dstream.h375 const DeviceMemory<BiasT> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm()
420 dnn::ActivationMode activation_mode) { in FusedConvolveRunnerFromDesc()
Ddnn.cc152 dnn::ActivationMode activation_mode, in GetFusedConvolveRunners()
168 dnn::ActivationMode activation_mode) { in FusedConvolveRunnerFromDesc()
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()
Dstream_executor_pimpl.cc324 dnn::ActivationMode activation_mode, in GetFusedConvolveRunners()
/external/tensorflow/tensorflow/core/kernels/
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
Dgpu_utils.cc189 double side_value_scale, se::dnn::ActivationMode activation_mode, in LogFusedConvForwardAutotuneResults()
Dconv_ops_gpu.cc115 const se::dnn::ActivationMode activation_mode, double conv_scale, in AutotuneFusedConv()
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()
/external/tensorflow/tensorflow/stream_executor/rocm/
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 …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc59 string activation_mode; in FusedBatchNormOp() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
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 …]
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc53 FusedBNActivationMode activation_mode; member
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1707 string activation_mode; in FusedBatchNormExRewrite() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3093 string data_format, filter_format, activation_mode, padding_type; in ConvertFusedConv2DBiasActivation() local