OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:Relu6GradOp
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
relu_op.cc
68
class
Relu6GradOp
: public XlaOpKernel {
70
explicit
Relu6GradOp
(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
90
REGISTER_XLA_OP(Name("Relu6Grad"),
Relu6GradOp
);
/external/tensorflow/tensorflow/core/kernels/
relu_op.h
105
class
Relu6GradOp
: public BinaryElementWiseOp<T,
Relu6GradOp
<Device, T>> {
107
using BinaryElementWiseOp<T,
Relu6GradOp
<Device, T>>::BinaryElementWiseOp;
125
void
Relu6GradOp
<Device, T>::OperateNoTemplate(OpKernelContext* context,
Completed in 300 milliseconds