Home
last modified time | relevance | path

Searched refs:activation_mode (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op_test_base.py209 strides, padding, activation_mode, data_format, argument
258 activation_mode=activation_mode)
302 activation_mode="Relu")
502 activation_mode="Relu")
513 activation_mode="Relu")
523 activation_mode="Relu")
533 activation_mode="Relu")
551 activation_mode="Relu"))
562 activation_mode="Relu"))
574 activation_mode="Tanh"))
[all …]
Dfused_conv2d_bias_activation_op.py38 activation_mode="Relu", argument
107 activation_mode=activation_mode,
Dfused_conv2d_bias_activation_benchmark.py114 activation_mode="Relu")
126 activation_mode="Relu")
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv_ops_gpu.h36 ActivationMode activation_mode) in FusedConvParameters() argument
39 activation_mode_(activation_mode), in FusedConvParameters()
42 hash_code_ = Hash64Combine(hash_code_, activation_mode); in FusedConvParameters()
Dfused_conv2d_bias_activation_op.h44 ActivationMode activation_mode, TensorFormat data_format,
58 ActivationMode activation_mode, TensorFormat data_format,
Dfused_conv2d_bias_activation_op.cc446 ActivationMode activation_mode, TensorFormat data_format, in launch() argument
677 activation_mode, in launch()
681 switch (activation_mode) { in launch()
689 LOG(FATAL) << "Activation mode " << activation_mode << " not supported"; in launch()
700 if (activation_mode == ActivationMode::NONE) { in launch()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h272 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
286 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
303 dnn::ActivationMode activation_mode,
318 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
412 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
540 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
553 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
564 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
573 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
584 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
[all …]
Drocm_dnn.cc821 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor() argument
832 switch (activation_mode) { in ScopedActivationDescriptor()
856 << dnn::ActivationModeString(activation_mode) in ScopedActivationDescriptor()
3153 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3170 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3188 dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3205 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3466 dnn::ActivationMode activation_mode, in DoActivate() argument
4029 const DeviceMemory<T>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivationImpl() argument
4049 ScopedActivationDescriptor activation_desc{activation_mode}; in DoFusedConvolutionBiasActivationImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1123 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
1140 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1159 dnn::ActivationMode activation_mode,
1178 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
1698 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode,
2359 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode,
2400 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
2413 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
2452 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data,
2465 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data,
[all …]
Dstream.cc448 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm() argument
457 PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), in ThenFusedConvolveWithAlgorithm()
466 activation_mode, output_descriptor, output, scratch_allocator, in ThenFusedConvolveWithAlgorithm()
486 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm() argument
495 PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), in ThenFusedConvolveWithAlgorithm()
504 activation_mode, output_descriptor, output, scratch_allocator, in ThenFusedConvolveWithAlgorithm()
525 dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm() argument
534 PARAM(bias_descriptor), PARAM(biases), PARAM(activation_mode), in ThenFusedConvolveWithAlgorithm()
543 activation_mode, output_descriptor, output, scratch_allocator, in ThenFusedConvolveWithAlgorithm()
563 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dstream.h345 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode,
359 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
374 dnn::ActivationMode activation_mode,
388 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode,
586 Stream &ThenActivate(dnn::ActivationMode activation_mode,
593 Stream &ThenActivateWithOptions(dnn::ActivationMode activation_mode,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h281 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode,
295 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
312 dnn::ActivationMode activation_mode,
327 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode,
421 bool DoActivate(Stream* stream, dnn::ActivationMode activation_mode,
591 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode,
Dcuda_dnn.cc798 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor() argument
804 switch (activation_mode) { in CudnnActivationDescriptor()
829 << static_cast<int>(activation_mode); in CudnnActivationDescriptor()
2930 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl() argument
2936 if (activation_mode != dnn::ActivationMode::kRelu && in DoFusedConvolveImpl()
2937 activation_mode != dnn::ActivationMode::kNone) { in DoFusedConvolveImpl()
2982 activation_mode, CUDNN_NOT_PROPAGATE_NAN, output_descriptor.value_max()); in DoFusedConvolveImpl()
3351 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3361 activation_mode, output_descriptor, output_data, in DoFusedConvolve()
3375 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto36 int64 activation_mode = 3; field
Dcudnn_conv_runner.cc373 if (!se::dnn::ActivationMode_IsValid(backend_config.activation_mode())) { in GetCudnnConvParams()
378 backend_config.activation_mode()); in GetCudnnConvParams()
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h449 const se::dnn::ActivationMode activation_mode)
450 : ConvParameters(base), activation_mode_(activation_mode) {}
/external/tensorflow/tensorflow/core/
DBUILD942 "util/activation_mode.h",