Searched refs:GatherNdShape (Results 1 – 4 of 4) sorted by relevance
264 Status GatherNdShape(InferenceContext* c);
2379 Status GatherNdShape(InferenceContext* c) { in GatherNdShape() function
318 .SetShapeFn(shape_inference::GatherNdShape);
1288 .SetShapeFn(shape_inference::GatherNdShape);