Searched refs:hlo_gather (Results 1 – 1 of 1) sorted by relevance
1244 auto hlo_gather = xla::Cast<xla::HloGatherInstruction>(instr); in EmitGatherOp() local1246 hlo_gather->gather_dimension_numbers(), &builder_)); in EmitGatherOp()1248 std::vector<int64_t>(hlo_gather->gather_slice_sizes().begin(), in EmitGatherOp()1249 hlo_gather->gather_slice_sizes().end()))); in EmitGatherOp()