Searched refs:shapeTensors (Results 1 – 1 of 1) sorted by relevance
9 std::vector<at::Tensor> shapeTensors; in getTensorShapes() local10 shapeTensors.reserve(tensors.size()); in getTensorShapes()16 shapeTensors.emplace_back(std::move(shapesTensor)); in getTensorShapes()18 return shapeTensors; in getTensorShapes()