Searched refs:BiasAddGradShape (Results 1 – 4 of 4) sorted by relevance
335 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()344 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()357 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()370 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()383 TF_EXPECT_OK(BiasAddGradShape(&c)); in TEST()392 EXPECT_FALSE(BiasAddGradShape(&c).ok()); in TEST()404 EXPECT_FALSE(BiasAddGradShape(&c).ok()); in TEST()
233 Status BiasAddGradShape(shape_inference::InferenceContext* c);
295 Status BiasAddGradShape(shape_inference::InferenceContext* c) { in BiasAddGradShape() function
254 .SetShapeFn(shape_inference::BiasAddGradShape);