MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ArgMinTwoPass

Function ArgMinTwoPass

tensorflow/compiler/xla/client/lib/arithmetic.cc:228–230  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

226}
227
228XlaOp ArgMinTwoPass(XlaOp input, PrimitiveType output_type, int axis) {
229 return ArgMinMaxTwoPass(input, output_type, axis, /*is_min=*/true);
230}
231} // namespace xla

Callers 1

CompileMethod · 0.85

Calls 1

ArgMinMaxTwoPassFunction · 0.85

Tested by

no test coverage detected