/external/tensorflow/tensorflow/core/util/ |
D | activation_mode.h | 30 enum ActivationMode { enum 42 ActivationMode* value);
|
D | activation_mode.cc | 24 ActivationMode* value) { in GetActivationModeFromString()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 264 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, 278 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, 326 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, 340 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, 355 dnn::ActivationMode activation_mode, 370 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, 464 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode, 592 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, 605 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, 616 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data, [all …]
|
D | rocm_dnn.cc | 938 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor() 950 case dnn::ActivationMode::kNone: in ScopedActivationDescriptor() 954 case dnn::ActivationMode::kSigmoid: in ScopedActivationDescriptor() 958 case dnn::ActivationMode::kRelu: in ScopedActivationDescriptor() 962 case dnn::ActivationMode::kRelu6: in ScopedActivationDescriptor() 967 case dnn::ActivationMode::kTanh: in ScopedActivationDescriptor() 3529 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward() 3549 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward() 3571 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl() 3683 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.cc | 95 std::string ActivationModeString(ActivationMode mode) { in ActivationModeString() 97 case ActivationMode::kNone: in ActivationModeString() 99 case ActivationMode::kSigmoid: in ActivationModeString() 101 case ActivationMode::kRelu: in ActivationModeString() 103 case ActivationMode::kRelu6: in ActivationModeString() 105 case ActivationMode::kReluX: in ActivationModeString() 107 case ActivationMode::kTanh: in ActivationModeString() 109 case ActivationMode::kBandPass: in ActivationModeString()
|
D | dnn.h | 927 std::string ActivationModeString(ActivationMode mode); 1019 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward() 1038 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward() 1151 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() 1169 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() 1189 dnn::ActivationMode activation_mode, in DoFusedConvolve() 1209 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() 1230 dnn::ActivationMode /*activation_mode*/, in DoFusedConvolve() argument 1764 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, in DoActivate() 2494 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation() [all …]
|
D | dnn.proto | 50 enum ActivationMode { enum
|
D | stream.h | 254 dnn::ActivationMode activation_mode, DeviceMemory<float> *y, 279 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y, 362 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, 376 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, 391 dnn::ActivationMode activation_mode, 405 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, 419 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, 609 Stream &ThenActivate(dnn::ActivationMode activation_mode, 616 Stream &ThenActivateWithOptions(dnn::ActivationMode activation_mode,
|
D | stream.cc | 57 std::string ToVlogString(dnn::ActivationMode mode) { in ToVlogString() 350 dnn::ActivationMode activation_mode, DeviceMemory<float> *y, in ThenBatchNormalizationForward() 401 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y, in ThenBatchNormalizationForward() 454 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() 485 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() 517 dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() 548 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() 579 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() 1017 Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode, in ThenActivate() 1025 Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode, in ThenActivateWithOptions()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 226 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, 240 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, 288 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, 302 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, 317 dnn::ActivationMode activation_mode, 332 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, 346 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, 440 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode, 597 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, 625 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode,
|
D | cuda_dnn.cc | 859 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor() 866 case dnn::ActivationMode::kNone: in CudnnActivationDescriptor() 869 case dnn::ActivationMode::kRelu6: in CudnnActivationDescriptor() 873 case dnn::ActivationMode::kReluX: in CudnnActivationDescriptor() 877 case dnn::ActivationMode::kRelu: in CudnnActivationDescriptor() 880 case dnn::ActivationMode::kSigmoid: in CudnnActivationDescriptor() 883 case dnn::ActivationMode::kTanh: in CudnnActivationDescriptor() 3185 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl() 3191 if (activation_mode != dnn::ActivationMode::kRelu && in DoFusedConvolveImpl() 3192 activation_mode != dnn::ActivationMode::kNone) { in DoFusedConvolveImpl() [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | attribute_exporter.h | 32 StatusOr<stream_executor::dnn::ActivationMode> ConvertConvActivationMode(
|
D | attribute_exporter.cc | 60 StatusOr<stream_executor::dnn::ActivationMode> ConvertConvActivationMode( in ConvertConvActivationMode()
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | conv_autotuning.proto | 26 stream_executor.dnn.ActivationMode activation = 8;
|
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_fused_impl.h | 310 const se::dnn::ActivationMode activation_mode) 322 std::tuple<ConvParameters::ParameterDataType, se::dnn::ActivationMode>; 329 se::dnn::ActivationMode activation_mode_; 563 se::dnn::ActivationMode dnn_activation_mode; 566 dnn_activation_mode = se::dnn::ActivationMode::kRelu;
|
D | gpu_utils.h | 238 double side_value_scale, se::dnn::ActivationMode activation_mode,
|
D | gpu_utils.cc | 178 double side_value_scale, se::dnn::ActivationMode activation_mode, in LogFusedConvForwardAutotuneResults()
|
D | fused_batch_norm_op.cc | 622 se::dnn::ActivationMode AsDnnActivationMode( in AsDnnActivationMode() 626 return se::dnn::ActivationMode::kNone; in AsDnnActivationMode() 628 return se::dnn::ActivationMode::kRelu; in AsDnnActivationMode()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_runner.h | 60 se::dnn::ActivationMode mode;
|
D | cudnn_batchnorm_runner.cc | 149 se::dnn::ActivationMode::kNone, // in RunCudnnBatchNormForwardInferenceImpl() 178 se::dnn::ActivationMode::kNone, // in RunCudnnBatchNormForwardTrainingImpl()
|
D | backend_configs.proto | 37 // stream_executor::dnn::ActivationMode.
|
D | cudnn_fused_conv_rewriter.cc | 248 static_cast<int64>(se::dnn::ActivationMode::kRelu)); in TryRewriteToCudnnForwardRelu()
|
D | gpu_conv_runner.cc | 309 static_cast<se::dnn::ActivationMode>(backend_config.activation_mode()); in GetGpuConvConfig()
|
D | ir_emitter_unnested.cc | 1187 TF_ASSIGN_OR_RETURN(stream_executor::dnn::ActivationMode activation_mode, in EmitConvolutionThunkFromMlir()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | mhlo_to_lhlo_with_xla.cc | 703 stream_executor::dnn::ActivationMode activation) { in GetLHLOActivation() 793 auto se_activation = static_cast<stream_executor::dnn::ActivationMode>( in EmitDnnConvolution()
|