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