Searched refs:size_splits_vector (Results 1 – 2 of 2) sorted by relevance
51 std::vector<int64_t>* size_splits_vector) { in GetSizeSplitsVector() argument54 size_splits_vector->push_back(GetTensorData<T>(size_splits)[i]); in GetSizeSplitsVector()67 std::vector<int64_t> size_splits_vector; in ResizeOutputTensors() local69 GetSizeSplitsVector<int32_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()71 GetSizeSplitsVector<int64_t>(size_splits, &size_splits_vector); in ResizeOutputTensors()80 for (int i = 0; i < size_splits_vector.size(); ++i) { in ResizeOutputTensors()81 if (size_splits_vector.at(i) == -1) { in ResizeOutputTensors()89 size_splits_sum += size_splits_vector.at(i); in ResizeOutputTensors()101 size_splits_vector[minus_one_index] = input_size - size_splits_sum; in ResizeOutputTensors()111 output_dims->data[axis_value] = size_splits_vector.at(i); in ResizeOutputTensors()
817 std::vector<int64> size_splits_vector; in ProcessTensorFlowSplitVOperator() local821 size_splits_vector.push_back(each_size); in ProcessTensorFlowSplitVOperator()824 size_splits_vector = size_array.GetBuffer<ArrayDataType::kInt64>().data; in ProcessTensorFlowSplitVOperator()847 CHECK_EQ(op->num_split, size_splits_vector.size()); in ProcessTensorFlowSplitVOperator()850 for (auto size : size_splits_vector) { in ProcessTensorFlowSplitVOperator()865 std::find(size_splits_vector.begin(), size_splits_vector.end(), -1); in ProcessTensorFlowSplitVOperator()876 (*output_shape.mutable_dims())[axis] = size_splits_vector.at(i); in ProcessTensorFlowSplitVOperator()