Searched defs:activation_mode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | backend_configs.proto | 36 int64 activation_mode = 3; field
|
| /external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
| D | fused_conv2d_bias_activation_op.py | 38 activation_mode="Relu", argument
|
| D | fused_conv2d_bias_activation_op_test_base.py | 209 strides, padding, activation_mode, data_format, argument
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv_ops_gpu.h | 36 ActivationMode activation_mode) in FusedConvParameters()
|
| D | fused_conv2d_bias_activation_op.cc | 446 ActivationMode activation_mode, TensorFormat data_format, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 821 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor()
3153 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3170 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3188 dnn::ActivationMode activation_mode, in DoFusedConvolve()
3205 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3466 dnn::ActivationMode activation_mode, in DoActivate()
4029 const DeviceMemory<T>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivationImpl()
4117 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation()
4136 double epsilon, dnn::ActivationMode activation_mode, in DoFusedBatchNormActivationInferenceImpl()
4213 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, in DoFusedBatchNormActivationInference()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 798 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor()
2930 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl()
3351 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3375 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3400 dnn::ActivationMode activation_mode, in DoFusedConvolve()
3424 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve()
3704 dnn::ActivationMode activation_mode, in DoActivate()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 448 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
486 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
525 dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
563 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm()
1438 Stream &Stream::ThenActivate(dnn::ActivationMode activation_mode, in ThenActivate()
1446 Stream &Stream::ThenActivateWithOptions(dnn::ActivationMode activation_mode, in ThenActivateWithOptions()
|