Searched refs:size_splits_sum (Results 1 – 2 of 2) sorted by relevance
78 int64_t size_splits_sum = 0; in ResizeOutputTensors() local89 size_splits_sum += size_splits_vector.at(i); in ResizeOutputTensors()96 if (size_splits_sum > input_size) { in ResizeOutputTensors()101 size_splits_vector[minus_one_index] = input_size - size_splits_sum; in ResizeOutputTensors()103 } else if (size_splits_sum != input_size) { in ResizeOutputTensors()
849 int64_t minus_one_count = 0, size_splits_sum = 0; in ProcessTensorFlowSplitVOperator() local854 size_splits_sum += size; in ProcessTensorFlowSplitVOperator()863 CHECK_LE(size_splits_sum, input_size); in ProcessTensorFlowSplitVOperator()866 *iter = input_size - size_splits_sum; in ProcessTensorFlowSplitVOperator()868 CHECK_EQ(size_splits_sum, input_size); in ProcessTensorFlowSplitVOperator()