Home
last modified time | relevance | path

Searched defs:leakyrelu_alpha (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_eigen_output_kernels.h54 float leakyrelu_alpha = 0.0; // Used by `LeakyRelu` fusion only member
136 float leakyrelu_alpha; member
158 float leakyrelu_alpha; member
H A Dconv_ops_gpu.cc116 double side_input_scale, double leakyrelu_alpha, in AutotuneFusedConv()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.cc1392 float leakyrelu_alpha; in MklFusedConvOp() local
1426 float leakyrelu_alpha; in MklFusedConvOp() local
1471 float epsilon, leakyrelu_alpha; in MklFusedConvOp() local
1511 float leakyrelu_alpha; in MklFusedConvOp() local
2161 float leakyrelu_alpha; in MklFusedConv3DOp() local
2196 float leakyrelu_alpha; in MklFusedConv3DOp() local
H A Dmkl_matmul_op_fused.cc321 float leakyrelu_alpha = 0.2; member in tensorflow::MklFusedMatMulOp
/aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/
H A Dconv_parameters.h47 double leakyrelu_alpha; member
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc765 float leakyrelu_alpha = 0.5; in RunTest() local
1613 float leakyrelu_alpha = 0.5; in RunTest() local
1826 float leakyrelu_alpha = 0.5; in TEST_F() local
2067 float leakyrelu_alpha = 0.5; in TEST_F() local
2424 auto leakyrelu_alpha = ops::Const<CType>(s.WithOpName("alpha"), epsilon); in RunTest() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dlazy_op_runner.h146 double conv_scale, side_input_scale, leakyrelu_alpha; member
H A Ddnn.cc146 double leakyrelu_alpha, Stream* stream, in GetFusedConvolveRunners()
162 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
H A Dstream.h414 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
H A Dstream_executor_pimpl.cc318 double leakyrelu_alpha, Stream* stream, in GetFusedConvolveRunners()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc3631 double alpha2, double leakyrelu_alpha, in GetCudnnFusedOperationGraph()
5066 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
5141 double leakyrelu_alpha, Stream* stream, in GetFusedConvolveRunners()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc2689 float leakyrelu_alpha; in CopyAttrsFromPadAndFusedConv2D() local