Searched refs:EluOp (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | elu_op.cc | 48 class EluOp : public XlaOpKernel { class
50 explicit EluOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluOp() function in tensorflow::__anonc10e36da0111::EluOp
74 REGISTER_XLA_OP(Name("Elu"), EluOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op.h | 196 class EluOp : public UnaryElementWiseOp<T, EluOp<Device, T>> {
198 using UnaryElementWiseOp<T, EluOp<Device, T>>::UnaryElementWiseOp;
|
| D | relu_op.cc | 60 EluOp<CPUDevice, type>); \
175 EluOp<GPUDevice, type>); \
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 57 return isa<EluOp, ReluOp, Relu6Op>(op); in IsActivationFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 132 TypeID::get<TF::EluOp>(), in IsOpAllowedTf2XlaFallback()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 1214 auto tf_elu_op = cast<TF::EluOp>(op); in matchAndRewrite()
|
| D | legalize_tfl.cc | 1890 auto tfl_elu_op = cast<TFL::EluOp>(op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | auto_mixed_precision_test.cc | 1074 TEST_F(AutoMixedPrecisionTest, EluOp) { in TEST_F() argument
|