Searched refs:BatchMatMulShape (Results 1 – 3 of 3) sorted by relevance
114 Status BatchMatMulShape(shape_inference::InferenceContext* c);
346 Status BatchMatMulShape(shape_inference::InferenceContext* c) { in BatchMatMulShape() function
127 .SetShapeFn(shape_inference::BatchMatMulShape);148 .SetShapeFn(shape_inference::BatchMatMulShape);