Searched defs:SetShapeFn (Results 1 – 3 of 3) sorted by relevance
274 OpDefBuilderWrapper& SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn() function
608 TEST_F(OpDefBuilderTest, SetShapeFn) { in TEST_F() argument
652 OpDefBuilder& OpDefBuilder::SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn() function in tensorflow::OpDefBuilder