Home
last modified time | relevance | path

Searched refs:leakyrelu_alpha (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_eigen_output_kernels.h53 float leakyrelu_alpha = 0.0; // Used by `LeakyRelu` fusion only member
120 static auto apply(XprType expr, const float leakyrelu_alpha) -> decltype(
127 leakyrelu_alpha)),
135 float leakyrelu_alpha; member
157 float leakyrelu_alpha; member
232 : bias_data(args.bias_add_data), leakyrelu_alpha(args.leakyrelu_alpha) {}
248 output = LeakyRelu::template apply<decltype(expr)>(expr, leakyrelu_alpha);
254 float leakyrelu_alpha;
308 leakyrelu_alpha(args.leakyrelu_alpha) {}
334 leakyrelu_alpha);
[all …]
Dfused_eigen_output_kernels.cc72 "leakyrelu_alpha", &fused_computation_args->leakyrelu_alpha)); in InitializeFusedComputation()
93 "leakyrelu_alpha", &fused_computation_args->leakyrelu_alpha)); in InitializeFusedComputation()
Dmatmul_op_fused.cc91 &fusion_args.leakyrelu_alpha)); in operator ()()
Dconv_ops_fused_impl.h226 &fusion_args.leakyrelu_alpha));
238 &fusion_args.leakyrelu_alpha));
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_matmul_op_fused.cc53 OP_REQUIRES_OK(ctx, ctx->GetAttr("leakyrelu_alpha", &leakyrelu_alpha)); in MklFusedMatMulOp()
282 {"leakyrelu", {1.0, leakyrelu_alpha, 0.0}}); in ExtendMklDnnMatMulFwdParams()
295 float leakyrelu_alpha = 0.2; member in tensorflow::MklFusedMatMulOp
Dmkl_conv_ops.cc1190 float leakyrelu_alpha; in MklFusedConvOp() local
1192 context->GetAttr("leakyrelu_alpha", &leakyrelu_alpha)); in MklFusedConvOp()
1194 leakyrelu_alpha); in MklFusedConvOp()
1216 float leakyrelu_alpha; in MklFusedConvOp() local
1218 context->GetAttr("leakyrelu_alpha", &leakyrelu_alpha)); in MklFusedConvOp()
1220 leakyrelu_alpha); in MklFusedConvOp()
1260 float leakyrelu_alpha; in MklFusedConvOp() local
1262 context->GetAttr("leakyrelu_alpha", &leakyrelu_alpha)); in MklFusedConvOp()
1264 leakyrelu_alpha); in MklFusedConvOp()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper_test.cc558 float leakyrelu_alpha = 0.5; in TEST_F() local
576 auto attr = ops::internal::LeakyRelu::Alpha(leakyrelu_alpha); in TEST_F()
619 EXPECT_EQ(node.attr().at("leakyrelu_alpha").f(), leakyrelu_alpha); in TEST_F()
661 float leakyrelu_alpha = 0.5; in RunTest() local
678 auto attr = ops::internal::LeakyRelu::Alpha(leakyrelu_alpha); in RunTest()
721 EXPECT_EQ(node.attr().at("leakyrelu_alpha").f(), leakyrelu_alpha); in RunTest()
857 float leakyrelu_alpha = 0.5; in TEST_F() local
870 auto attr = ops::internal::LeakyRelu::Alpha(leakyrelu_alpha); in TEST_F()
921 EXPECT_EQ(node.attr().at("leakyrelu_alpha").f(), leakyrelu_alpha); in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc2712 float leakyrelu_alpha; in CopyAttrsFromPadAndFusedConv2D() local
2724 GetNodeAttr(fused_conv2d->def(), "leakyrelu_alpha", &leakyrelu_alpha)); in CopyAttrsFromPadAndFusedConv2D()
2737 nb->Attr("leakyrelu_alpha", leakyrelu_alpha); in CopyAttrsFromPadAndFusedConv2D()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td18365 DefaultValuedAttr<F32Attr, "0.2f">:$leakyrelu_alpha
18410 DefaultValuedAttr<F32Attr, "0.2f">:$leakyrelu_alpha