Home
last modified time | relevance | path

Searched defs:dimsB (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DExpandUtils.cpp20 auto dimsB = static_cast<ptrdiff_t>(b.size()); in infer_size_impl() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h302 const T* bufferB, const std::vector<uint32_t>& dimsB, T* merged) { in mergeThirdDimension()