Searched defs:ElementwiseInterpreter (Results 1 – 1 of 1) sorted by relevance
244 struct ElementwiseInterpreter : torch::CustomClassHolder { struct245 using InstructionType = std::tuple<251 ElementwiseInterpreter() {} in ElementwiseInterpreter() argument257 void setInstructions(std::vector<InstructionType> instructions) { in setInstructions()264 void addConstant(const std::string& name, at::Tensor value) { in addConstant()272 void setInputNames(std::vector<std::string> input_names) { in setInputNames()279 void setOutputName(std::string output_name) { in setOutputName()285 at::Tensor __call__(std::vector<at::Tensor> inputs) { in __call__()359 using SerializationType = std::tuple<367 SerializationType __getstate__() const { in __getstate__()[all …]