Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp1780 tensorOutputStrides_.emplace_back(); in compile()
1818 tensorOutputStrides_.push_back(*strides); in compile()
1820 tensorOutputStrides_.push_back(TensorType::contiguousStridesOf(sizes)); in compile()
1842 tensorOutputStrides_.resize(bufOutputs_.size()); in compile()
1924 *strides = tensorOutputStrides_; in getStaticOutputSizesAndStrides()
2005 tensorOutputStrides_[i], in prepareRunArgs()
H A Dkernel.h308 std::vector<std::vector<int64_t>> tensorOutputStrides_; variable