Home
last modified time | relevance | path

Searched refs:tensorIdToValue_ (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/torch/csrc/jit/codegen/onednn/
Dkernel.cpp27 tensorIdToValue_ = llgaGraphHelper.getTensorIdToValue(); in LlgaKernel()
47 tensorIdToValue_.count(inputId) > 0, in initializeConstantInputs()
51 auto* value = tensorIdToValue_[inputId]; in initializeConstantInputs()
Dkernel.h72 std::map<size_t, Value*> tensorIdToValue_; variable
Dgraph_helper.h84 tensorIdToValue_; // map from tensorId to torch::jit::Value variable
Dgraph_helper.cpp425 tensorIdToValue_.emplace(input->unique(), input); in LlgaGraphHelper()
583 return tensorIdToValue_; in getTensorIdToValue()