Searched defs:AllToAllOp (Results 1 – 2 of 2) sorted by relevance
68 class AllToAllOp : public XlaOpKernel { class70 explicit AllToAllOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in AllToAllOp() function in tensorflow::__anon464e947c0111::AllToAllOp
1142 static StringRef CustomCallTarget(AllToAllOp) { return "xla.gpu.all_to_all"; } in CustomCallTarget() argument