Searched refs:OpShapeInferenceFn (Results 1 – 4 of 4) sorted by relevance
37 OpShapeInferenceFn; typedef43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,48 OpShapeInferenceFn shape_inference_fn;
36 static OpShapeInferenceFn* global_fn_ptr = nullptr;52 const string& expected_outs, OpShapeInferenceFn fn) { in RunInferShapes()
44 OpShapeInferenceFn* shape_fn_out = nullptr) { in ExpectSuccess()
630 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()