Searched refs:batch_dim_bounds (Results 1 – 1 of 1) sorted by relevance
89 std::vector<int64> batch_dim_bounds; in AdjustBatchDimsInAccumulator() local90 batch_dim_bounds.reserve(start_indices_shape.dimensions_size()); in AdjustBatchDimsInAccumulator()93 batch_dim_bounds.push_back(start_indices_shape.dimensions(i)); in AdjustBatchDimsInAccumulator()97 if (batch_dim_bounds.empty()) { in AdjustBatchDimsInAccumulator()105 return ExpandFirstDimIntoNDims(accumulator, batch_dim_bounds); in AdjustBatchDimsInAccumulator()