Searched refs:BiasAddGradShape (Results 1 – 4 of 4) sorted by relevance
553 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()562 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()575 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()588 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()601 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()609 EXPECT_FALSE(BiasAddGradShape(&c).ok()); in TEST()621 EXPECT_FALSE(BiasAddGradShape(&c).ok()); in TEST()
123 Status BiasAddGradShape(shape_inference::InferenceContext* c);
441 Status BiasAddGradShape(shape_inference::InferenceContext* c) { in BiasAddGradShape() function
320 .SetShapeFn(shape_inference::BiasAddGradShape);