Home
last modified time | relevance | path

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

/external/pytorch/test/cpp/tensorexpr/
Dtest_cuda.cpp732 ExprHandle a_kmn = Load::make(a, {k * (M * N) + m * N + n}); in TEST() local
867 ExprHandle a_kmn = a.load(k * (M * N) + m * N + n); in TEST() local