Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.h254 Tensor convertStaticShapeOutputToCorrectStrides(torch::jit::Value* v);
H A Dkernel.cpp1343 Tensor TensorExprKernel::convertStaticShapeOutputToCorrectStrides( in convertStaticShapeOutputToCorrectStrides() function in TensorExprKernel
1806 convertStaticShapeOutputToCorrectStrides(output); in compile()