Searched refs:EluGradOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | elu_op.cc | 42 class EluGradOp : public XlaOpKernel { class
44 explicit EluGradOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in EluGradOp() function in tensorflow::__anonb35316230111::EluGradOp
60 REGISTER_XLA_OP(Name("EluGrad"), EluGradOp);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | relu_op.h | 208 class EluGradOp : public BinaryElementWiseOp<T, EluGradOp<Device, T>> {
210 using BinaryElementWiseOp<T, EluGradOp<Device, T>>::BinaryElementWiseOp;
228 void EluGradOp<Device, T>::OperateNoTemplate(OpKernelContext* context, in OperateNoTemplate()
|
| D | relu_op.cc | 65 EluGradOp<CPUDevice, type>); \
180 EluGradOp<GPUDevice, type>); \
240 EluGradOp<SYCLDevice, type>); \
|