Searched defs:TorchScriptIrBuilder (Results 1 – 1 of 1) sorted by relevance
16 struct TorchScriptIrBuilder : IrBuilder { struct17 NodePtr MakeDeviceData( in MakeDeviceData()23 NodePtr MakeScalar(const at::Scalar& value, const at::ScalarType& type) in MakeScalar()27 NodePtr MakeExpand( in MakeExpand()40 NodePtr MakeTensorList(const OpList& inputs) const override { in MakeTensorList()56 NodePtr MakeSizeNode(const Value& input, size_t dim) const override { in MakeSizeNode()59 NodePtr MakeSizeAdd(const Value& a, const Value& b) const override { in MakeSizeAdd()62 NodePtr MakeSizeMul(const Value& a, const Value& b) const override { in MakeSizeMul()65 NodePtr MakeSizeDiv(const Value& a, const Value& b) const override { in MakeSizeDiv()