Searched refs:RandomShape (Results 1 – 5 of 5) sorted by relevance
34 .SetShapeFn(shape_inference::RandomShape);60 return shape_inference::RandomShape(c); in __anon2645a1db0102()71 .SetShapeFn(shape_inference::RandomShape);92 return shape_inference::RandomShape(c); in __anon2645a1db0202()103 .SetShapeFn(shape_inference::RandomShape);
300 .SetShapeFn(shape_inference::RandomShape);
244 Status RandomShape(shape_inference::InferenceContext* c);
2082 Status RandomShape(shape_inference::InferenceContext* c) { in RandomShape() function
921 TEST(XlaCompilationTest, RandomShape) { in TEST() argument