SetInferShape
Function Usage
Registers the shape inference function.
Prototype
OpDef &SetInferShape(gert::OpImplRegisterV2::InferShapeKernelFunc func);
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
func |
Input |
Shape inference function. The InferShapeKernelFunc type is defined as follows: using InferShapeKernelFunc = UINT32 (*)(InferShapeContext *); |
Returns
OpDef operator definition. For details, see OpDef.
Constraints
None
Parent topic: OpDef