Searched defs:ArgMaxOp (Results 1 – 1 of 1) sorted by relevance
115 class ArgMaxOp class118 explicit ArgMaxOp(OpKernelConstruction* context) in ArgMaxOp() function in tensorflow::ArgMaxOp