Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A DLlgaTensorImpl.h213 void set_input_tensor_index(size_t index) { in set_input_tensor_index() function
H A Dkernel.cpp241 outputSpecs_[outputOffset].set_input_tensor_index(inputOffset); in compile()