Searched refs:NnapiCompilation (Results 1 – 4 of 4) sorted by relevance
13 m.class_<torch::nnapi::bind::NnapiCompilation>("Compilation") in TORCH_LIBRARY()15 .def("init", &torch::nnapi::bind::NnapiCompilation::init) in TORCH_LIBRARY()16 .def("init2", &torch::nnapi::bind::NnapiCompilation::init2) in TORCH_LIBRARY()17 .def("run", &torch::nnapi::bind::NnapiCompilation::run) in TORCH_LIBRARY()
35 struct NnapiCompilation : torch::jit::CustomClassHolder { struct36 NnapiCompilation() = default;37 ~NnapiCompilation() override = default;
36 void NnapiCompilation::init( in init()47 void NnapiCompilation::init2( in init2()114 void NnapiCompilation::run( in run()166 void NnapiCompilation::get_operand_type(const at::Tensor& t, ANeuralNetworksOperandType* operand, s… in get_operand_type()
105 std::unique_ptr<torch::nnapi::bind::NnapiCompilation> comp_;126 comp_ = std::make_unique<torch::nnapi::bind::NnapiCompilation>(); in init()