OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:ReluGradOp
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
relu_op.cc
53
class
ReluGradOp
: public XlaOpKernel {
55
explicit
ReluGradOp
(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
89
REGISTER_XLA_OP(Name("ReluGrad"),
ReluGradOp
);
/external/tensorflow/tensorflow/core/kernels/
relu_op.h
62
class
ReluGradOp
: public BinaryElementWiseOp<T,
ReluGradOp
<Device, T>> {
64
using BinaryElementWiseOp<T,
ReluGradOp
<Device, T>>::BinaryElementWiseOp;
83
void
ReluGradOp
<Device, T>::OperateNoTemplate(OpKernelContext* context,
Completed in 86 milliseconds