Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/tensorexpr/
Dkernel.h227 void getStaticOutputSizesAndStrides(
Dkernel.cpp1910 void TensorExprKernel::getStaticOutputSizesAndStrides( in getStaticOutputSizesAndStrides() function in TensorExprKernel
1980 getStaticOutputSizesAndStrides(inputs, &static_sizes, &static_strides); in prepareRunArgs()
2096 getStaticOutputSizesAndStrides( in runWithAllocatedOutputs()