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

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
index_ops.h 26 class XlaArgMinMaxOp : public XlaOpKernel {
28 explicit XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min);
35 class XlaArgMaxOp : public XlaArgMinMaxOp {
index_ops.cc 34 XlaArgMinMaxOp::XlaArgMinMaxOp(OpKernelConstruction* ctx, bool is_min)
37 void XlaArgMinMaxOp::Compile(XlaOpKernelContext* ctx) {
82 : XlaArgMinMaxOp(ctx, /*is_min=*/false) {}
90 class XlaArgMinOp : public XlaArgMinMaxOp {
95 : XlaArgMinMaxOp(ctx, /*is_min=*/true) {}

Completed in 187 milliseconds