Searched defs:SetShapeFn (Results 1 – 3 of 3) sorted by relevance
264 OpDefBuilderWrapper<true>& SetShapeFn( in SetShapeFn() function289 OpDefBuilderWrapper<false>& SetShapeFn( in SetShapeFn() function
608 TEST_F(OpDefBuilderTest, SetShapeFn) { in TEST_F() argument
626 OpDefBuilder& OpDefBuilder::SetShapeFn(OpShapeInferenceFn fn) { in SetShapeFn() function in tensorflow::OpDefBuilder