Searched defs:MatMulShape (Results 1 – 2 of 2) sorted by relevance
116 struct MatMulShape { struct117 int64 m;118 int64 k;119 int64 n;
213 Status MatMulShape(shape_inference::InferenceContext* c) { in MatMulShape() function