hidet.ir

Submodules

• hidet.ir.type
  • DataType
  • data_type()
• hidet.ir.expr
  • IfThenElse
  • Int
  • if_then_else()
  • tensor_slice()
• hidet.ir.stmt
  • DeclareScope
• hidet.ir.func
  • Function
    • Function.get_attr()
• hidet.ir.compute
  • ReduceType
  • scalar_input()
  • tensor_input()
  • compute()
  • reduce()
  • arg_reduce()
• hidet.ir.task
  • Task
    • Task.name
    • Task.inputs
    • Task.outputs
    • Task.inverse_map
    • Task.attrs
    • Task.assertions
    • Task.share_map
    • Task.symbols
    • Task.generate_arguments()
    • Task.dummy_arguments()
    • Task.build()