Expand description
Output-shape and output-dtype inference for StdTensorOp.
Called during StdTensorOp -> ExecProgram lowering to populate
ExecInstruction::output_shapes and ExecInstruction::dtype.
Functionsยง
- infer_
output_ dtype - Infer output dtype for a single instruction given its op and input dtypes.
- infer_
output_ shapes - Infer output shapes for a single instruction.