Searched refs:variable_tensor_vec (Results 1 – 1 of 1) sorted by relevance
1959 std::vector<int> variable_tensor_vec; in InferShapesWithResourceVar() local1961 variable_tensor_vec.reserve(variable_tensor->dims()); in InferShapesWithResourceVar()1963 variable_tensor_vec.push_back(variable_tensor->dim_size(d)); in InferShapesWithResourceVar()1968 variable_tensor_vec.data(), variable_tensor_vec.size(), in InferShapesWithResourceVar()