Searched refs:gather_instr (Results 1 – 1 of 1) sorted by relevance
308 HloInstruction* gather_instr) { in ExpandInstruction() argument309 CHECK(!ShapeUtil::IsZeroElementArray(gather_instr->shape())); in ExpandInstruction()311 HloComputation* computation = gather_instr->parent(); in ExpandInstruction()312 HloInstruction* operand = gather_instr->mutable_operand(0); in ExpandInstruction()313 HloInstruction* start_indices = gather_instr->mutable_operand(1); in ExpandInstruction()315 const Shape& output_shape = gather_instr->shape(); in ExpandInstruction()319 gather_instr->gather_dimension_numbers(); in ExpandInstruction()332 gather_instr->ToString()); in ExpandInstruction()344 gather_instr->gather_slice_sizes(), gather_loop_trip_count, in ExpandInstruction()345 gather_instr->gather_dimension_numbers()); in ExpandInstruction()[all …]