Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cu.cc128 bool add_side_input, FusedBatchNormActivationMode activation_mode,
164 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
166 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
177 FusedBatchNormActivationMode activation_mode>
179 add_side_input, activation_mode,
193 activation_mode,
247 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
250 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
279 if (activation_mode == FusedBatchNormActivationMode::kIdentity) { in run()
282 } else if (activation_mode == FusedBatchNormActivationMode::kRelu) { in run()
[all …]
Dfused_batch_norm_op.h33 std::string ToString(FusedBatchNormActivationMode activation_mode);
36 FusedBatchNormActivationMode* activation_mode);
57 FusedBatchNormActivationMode activation_mode,
Dfused_batch_norm_ex_op_test.cc141 bool has_side_input, const string& activation_mode, in RunFusedBatchNorm() argument
178 if (activation_mode == "Relu") { in RunFusedBatchNorm()
187 if (activation_mode == "Relu") { in RunFusedBatchNorm()
230 bool has_side_input, const string& activation_mode, in RunFusedBatchNormEx() argument
277 .Attr("activation_mode", activation_mode) in RunFusedBatchNormEx()
283 if (activation_mode == "Relu") { in RunFusedBatchNormEx()
430 const string& activation_mode) { in VerifyFusedBatchNormEx() argument
440 has_side_input, activation_mode, fwd, bwd); in VerifyFusedBatchNormEx()
452 has_side_input, activation_mode, fwd, bwd); in VerifyFusedBatchNormEx()
572 string activation_mode = side_input_and_activation.second; in FusedBatchNormEx() local
[all …]
Dfused_batch_norm_op.cc57 string ToString(FusedBatchNormActivationMode activation_mode) { in ToString() argument
58 switch (activation_mode) { in ToString()
67 FusedBatchNormActivationMode* activation_mode) { in ParseActivationMode() argument
72 *activation_mode = FusedBatchNormActivationMode::kIdentity; in ParseActivationMode()
76 *activation_mode = FusedBatchNormActivationMode::kRelu; in ParseActivationMode()
98 FusedBatchNormActivationMode activation_mode, in operator ()()
108 activation_mode == FusedBatchNormActivationMode::kIdentity, in operator ()()
236 FusedBatchNormActivationMode activation_mode, in operator ()()
246 activation_mode == FusedBatchNormActivationMode::kIdentity, in operator ()()
623 const FusedBatchNormActivationMode activation_mode) { in AsDnnActivationMode() argument
[all …]
Dgpu_utils.cc178 double side_value_scale, se::dnn::ActivationMode activation_mode, in LogFusedConvForwardAutotuneResults() argument
190 instr.set_activation(activation_mode); in LogFusedConvForwardAutotuneResults()
Dgpu_utils.h238 double side_value_scale, se::dnn::ActivationMode activation_mode,
Dconv_ops_fused_impl.h310 const se::dnn::ActivationMode activation_mode)
311 : ConvParameters(base), activation_mode_(activation_mode) {}
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h264 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 …]
Drocm_dnn.cc938 ScopedActivationDescriptor(dnn::ActivationMode activation_mode) in ScopedActivationDescriptor() argument
949 switch (activation_mode) { in ScopedActivationDescriptor()
973 << dnn::ActivationModeString(activation_mode) in ScopedActivationDescriptor()
3529 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward() argument
3537 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3549 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward() argument
3557 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3571 dnn::ActivationMode activation_mode, DeviceMemory<T>* y, in DoBatchNormalizationForwardImpl() argument
3683 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
3699 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc55 string activation_mode; in FusedBatchNormOp() local
56 OP_REQUIRES_OK(ctx, ctx->GetAttr("activation_mode", &activation_mode)); in FusedBatchNormOp()
58 activation_mode == "Identity" || activation_mode == "Relu", in FusedBatchNormOp()
61 activation_mode)); in FusedBatchNormOp()
62 apply_relu_ = (activation_mode == "Relu"); in FusedBatchNormOp()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h226 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,
Dcuda_dnn.cc859 CudnnActivationDescriptor(dnn::ActivationMode activation_mode, in CudnnActivationDescriptor() argument
865 switch (activation_mode) { in CudnnActivationDescriptor()
888 << static_cast<int>(activation_mode); in CudnnActivationDescriptor()
3185 const DeviceMemory<BiasType>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolveImpl() argument
3191 if (activation_mode != dnn::ActivationMode::kRelu && in DoFusedConvolveImpl()
3192 activation_mode != dnn::ActivationMode::kNone) { in DoFusedConvolveImpl()
3240 activation_mode, CUDNN_NOT_PROPAGATE_NAN, output_descriptor.value_max()); in DoFusedConvolveImpl()
3455 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward() argument
3465 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in DoBatchNormalizationForward()
3478 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1019 dnn::ActivationMode activation_mode, DeviceMemory<float>* y, in DoBatchNormalizationForward() argument
1038 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y, in DoBatchNormalizationForward() argument
1151 const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
1169 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
1189 dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
1209 const DeviceMemory<float>& biases, dnn::ActivationMode activation_mode, in DoFusedConvolve() argument
1764 virtual bool DoActivate(Stream* stream, ActivationMode activation_mode, in DoActivate() argument
2494 const DeviceMemory<float>& bias_data, dnn::ActivationMode activation_mode, in DoFusedConvolutionBiasActivation() argument
2535 dnn::ActivationMode activation_mode, DeviceMemory<float>* y_data, in DoFusedBatchNormActivationInference() argument
2548 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half>* y_data, in DoFusedBatchNormActivationInference() argument
[all …]
Dstream.cc350 dnn::ActivationMode activation_mode, DeviceMemory<float> *y, in ThenBatchNormalizationForward() argument
362 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in ThenBatchNormalizationForward()
401 dnn::ActivationMode activation_mode, DeviceMemory<Eigen::half> *y, in ThenBatchNormalizationForward() argument
413 activation_mode, y, batch_mean, batch_var, saved_mean, saved_inv_var, in ThenBatchNormalizationForward()
454 const DeviceMemory<double> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() argument
463 PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), in FusedConvolveWithAlgorithm()
470 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
485 const DeviceMemory<float> &biases, dnn::ActivationMode activation_mode, in FusedConvolveWithAlgorithm() argument
494 PARAM(activation_mode), PARAM(output_descriptor), PARAM(output), in FusedConvolveWithAlgorithm()
501 side_input_scale, bias_descriptor, biases, activation_mode, in FusedConvolveWithAlgorithm()
[all …]
Dstream.h254 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,
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dlhlo_gpu_ops.td143 ActivationAttr:$activation_mode)>.attributes);
159 ActivationAttr:$activation_mode,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbackend_configs.proto38 int64 activation_mode = 3; field
Dgpu_conv_runner.cc304 if (!se::dnn::ActivationMode_IsValid(backend_config.activation_mode())) { in GetGpuConvConfig()
309 static_cast<se::dnn::ActivationMode>(backend_config.activation_mode()); in GetGpuConvConfig()
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo_gpu_ops.mlir127 {activation_mode = "Relu",
158 {activation_mode = "Relu",
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dhlo_text_to_lhlo_no_opt.hlotxt307 …rithm\":\"2\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"0\",\"sid…
314 // CHECK-SAME: activation_mode = "Relu"
346 …rithm\":\"0\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"sid…
353 // CHECK-SAME: activation_mode = "Relu"
386 …rithm\":\"0\",\"tensor_ops_enabled\":false,\"conv_result_scale\":1,\"activation_mode\":\"2\",\"sid…
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc48 FusedBNActivationMode activation_mode; member
53 FusedBNActivationMode activation_mode) in MklBatchNormFwdParams()
58 activation_mode(activation_mode), in MklBatchNormFwdParams()
193 if (fwdParams.activation_mode == FusedBNActivationMode::kRelu) { in Setup()
359 key_creator.AddAsKey<FusedBNActivationMode>(fwdParams.activation_mode); in CreateKey()
785 if (fwdParams.activation_mode == FusedBNActivationMode::kRelu) { in Compute()
/external/tensorflow/tensorflow/core/util/
DBUILD141 "activation_mode.h",
209 "activation_mode.cc",
331 "activation_mode.h",
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1755 string activation_mode; in FusedBatchNormExRewrite() local
1756 TF_CHECK_OK(GetNodeAttr(n->def(), "activation_mode", &activation_mode)); in FusedBatchNormExRewrite()
1766 if (activation_mode != "Relu") { in FusedBatchNormExRewrite()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3560 const auto activation_mode = attrs.get<string>("activation_mode"); in ConvertFusedConv2DBiasActivation() local
3561 auto op_pair = ActivationTypeMap()->find(activation_mode); in ConvertFusedConv2DBiasActivation()
3562 if (op_pair == ActivationTypeMap()->end() && activation_mode != "None") { in ConvertFusedConv2DBiasActivation()
3563 return errors::Unimplemented("Activation mode: ", activation_mode, in ConvertFusedConv2DBiasActivation()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc794 backend_config.activation_mode()); in EmitDnnConvolution()

12