Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/codegen/onednn/
Dkernel.h63 static dnnl::graph::logical_tensor toLogicalTensor(const ArgSpec& s) { in toLogicalTensor() function
Dkernel.cpp213 auto inputs = fmap(inputSpecs_, toLogicalTensor); in compile()
214 auto outputs = fmap(outputSpecs_, toLogicalTensor); in compile()