HomeSort by relevance Sort by last modified time
    Searched defs:Relu6Op (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
relu_op.cc 41 class Relu6Op : public XlaOpKernel {
43 explicit Relu6Op(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {}
88 REGISTER_XLA_OP(Name("Relu6"), Relu6Op);
  /external/tensorflow/tensorflow/core/kernels/
relu_op.h 93 class Relu6Op : public UnaryElementWiseOp<T, Relu6Op<Device, T>> {
95 using UnaryElementWiseOp<T, Relu6Op<Device, T>>::UnaryElementWiseOp;
114 // a (inputs): inputs that were passed to Relu6Op()

Completed in 57 milliseconds