Searched refs:gather_loop_trip_count (Results 1 – 1 of 1) sorted by relevance
234 absl::Span<const int64> slice_sizes, int64 gather_loop_trip_count, in CreateGatherLoopAccumulatorInitValue() argument238 accumulator_state_shape_dims.push_back(gather_loop_trip_count); in CreateGatherLoopAccumulatorInitValue()321 int64 gather_loop_trip_count = 1; in ExpandInstruction() local324 gather_loop_trip_count *= start_indices_shape.dimensions(i); in ExpandInstruction()328 if (!IsInt32(gather_loop_trip_count)) { in ExpandInstruction()339 CHECK_EQ(gather_loop_trip_count, in ExpandInstruction()344 gather_instr->gather_slice_sizes(), gather_loop_trip_count, in ExpandInstruction()349 computation, gather_loop_trip_count, in ExpandInstruction()