| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 38 int64 activation_mode = 3; field
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 141 bool has_side_input, const string& activation_mode, in RunFusedBatchNorm()
230 bool has_side_input, const string& activation_mode, in RunFusedBatchNormEx()
430 const string& activation_mode) { in VerifyFusedBatchNormEx()
572 string activation_mode = side_input_and_activation.second; in FusedBatchNormEx() local
|
| D | gpu_utils.cc | 178 double side_value_scale, se::dnn::ActivationMode activation_mode, in LogFusedConvForwardAutotuneResults()
|
| D | fused_batch_norm_op.cc | 57 string ToString(FusedBatchNormActivationMode activation_mode) { in ToString()
67 FusedBatchNormActivationMode* activation_mode) { in ParseActivationMode()
623 const FusedBatchNormActivationMode activation_mode) { in AsDnnActivationMode()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 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()
1764 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, in DoActivate()
2494 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()
2535 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, in DoFusedBatchNormActivationInference()
2548 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data, in DoFusedBatchNormActivationInference()
[all …]
|
| D | stream.cc | 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/rocm/ |
| D | rocm_dnn.cc | 938 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) 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()
3699 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3716 dnn::ActivationMode activation_mode, in DoFusedConvolve()
3732 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3992 dnn::ActivationMode activation_mode, in DoActivate()
4625 const DeviceMemory<T>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivationImpl()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 55 string activation_mode; in FusedBatchNormOp() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 859 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor()
3185 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl()
3455 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward()
3478 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward()
3503 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl()
3764 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3786 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3809 dnn::ActivationMode activation_mode, in DoFusedConvolve()
3831 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3862 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 48 FusedBNActivationMode activation_mode; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1755 string activation_mode; in FusedBatchNormExRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 3560 const auto activation_mode = attrs.get<string>("activation_mode"); in ConvertFusedConv2DBiasActivation() local
|