Searched defs:leakyrelu_alpha (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_eigen_output_kernels.h | 54 float leakyrelu_alpha = 0.0; // Used by `LeakyRelu` fusion only member
136 float leakyrelu_alpha; member
158 float leakyrelu_alpha; member
|
| D | conv_ops_gpu.cc | 116 double side_input_scale, double leakyrelu_alpha, in AutotuneFusedConv()
|
| /external/tensorflow/tensorflow/core/util/autotune_maps/ |
| D | conv_parameters.h | 47 double leakyrelu_alpha; member
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | lazy_op_runner.h | 146 double conv_scale, side_input_scale, leakyrelu_alpha; member
|
| D | dnn.cc | 146 double leakyrelu_alpha, Stream* stream, in GetFusedConvolveRunners()
162 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
|
| D | stream.h | 414 double side_input_scale, double leakyrelu_alpha, in FusedConvolveRunnerFromDesc()
|
| D | stream_executor_pimpl.cc | 318 double leakyrelu_alpha, Stream* stream, in GetFusedConvolveRunners()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 321 float leakyrelu_alpha = 0.2; member in tensorflow::MklFusedMatMulOp
|
| D | mkl_conv_ops.cc | 1392 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
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 765 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
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3631 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 2689 float leakyrelu_alpha; in CopyAttrsFromPadAndFusedConv2D() local
|