Home
last modified time | relevance | path

Searched refs:tensorOutputSymbolicSizes_ (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.h328 std::vector<std::vector<ExprHandle>> tensorOutputSymbolicSizes_; variable
H A Dkernel.cpp1789 tensorOutputSymbolicSizes_.push_back(sizes); in compile()
1919 tensorOutputSymbolicSizes_.size() == bufOutputs_.size()); in getStaticOutputSizesAndStrides()
1929 for (auto t : tensorOutputSymbolicSizes_[i]) { in getStaticOutputSizesAndStrides()