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