Searched refs:OpShapeInferenceFn (Results 1 – 6 of 6) sorted by relevance
61 OpShapeInferenceFn; typedef67 OpRegistrationData(const OpDef& def, const OpShapeInferenceFn& fn,72 OpShapeInferenceFn shape_inference_fn;239 OpDefBuilder& SetShapeFn(OpShapeInferenceFn fn);
35 static OpShapeInferenceFn* global_fn_ptr = nullptr;51 const string& expected_outs, OpShapeInferenceFn fn) { in RunInferShapes()
274 OpDefBuilderWrapper& SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()
652 OpDefBuilder& OpDefBuilder::SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn()657 op_reg_data_.shape_inference_fn = OpShapeInferenceFn(fn); in SetShapeFn()
44 OpShapeInferenceFn* shape_fn_out = nullptr) { in ExpectSuccess()
28 using ::tensorflow::OpShapeInferenceFn;