Searched refs:OpShapeInferenceFn (Results 1 – 6 of 6) sorted by relevance
37 OpShapeInferenceFn; typedef43 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,48 OpShapeInferenceFn shape_inference_fn;143 OpDefBuilder& SetShapeFn(OpShapeInferenceFn fn);
35 static OpShapeInferenceFn* global_fn_ptr = nullptr;51 const string& expected_outs, OpShapeInferenceFn fn) { in RunInferShapes()
262 OpDefBuilderWrapper& SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()
628 OpDefBuilder& OpDefBuilder::SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()633 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
44 OpShapeInferenceFn* shape_fn_out = nullptr) { in ExpectSuccess()
28 using ::tensorflow::OpShapeInferenceFn;