Home
last modified time | relevance | path

Searched refs:leakyrelu (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcwise_op_leakyrelu_bf16.cc72 struct leakyrelu : base<T, Eigen::internal::leakyrelu_op<T>> {}; struct
110 REGISTER(LeakyReluOp, CPU, "LeakyRelu", functor::leakyrelu, bfloat16);
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc283 csinfo_.leakyrelu = "LeakyRelu"; in MklLayoutRewritePass()
528 rinfo_.push_back({csinfo_.leakyrelu, in MklLayoutRewritePass()
529 mkl_op_registry::GetMklOpName(csinfo_.leakyrelu), in MklLayoutRewritePass()
891 string leakyrelu; member