Searched refs:RandomShape (Results 1 – 4 of 4) sorted by relevance
34 .SetShapeFn(shape_inference::RandomShape);50 return shape_inference::RandomShape(c); in __anon979c5d7d0102()61 .SetShapeFn(shape_inference::RandomShape);82 return shape_inference::RandomShape(c); in __anon979c5d7d0202()93 .SetShapeFn(shape_inference::RandomShape);
310 Status RandomShape(shape_inference::InferenceContext* c);
1380 Status RandomShape(shape_inference::InferenceContext* c) { in RandomShape() function
789 TEST(XlaCompilationTest, RandomShape) { in TEST() argument