Searched defs:gather_output (Results 1 – 1 of 1) sorted by relevance
39 xla::XlaOp* gather_output) { in XlaGather()158 int batch_dims, xla::XlaOp* gather_output) { in XlaGatherWithBatchDimsOpImpl()