Searched defs:CrossReplicaSumOp (Results 1 – 1 of 1) sorted by relevance
51 class CrossReplicaSumOp : public XlaOpKernel { class53 explicit CrossReplicaSumOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in CrossReplicaSumOp() function in tensorflow::__anonef304f720111::CrossReplicaSumOp