Searched defs:MatMulShape (Results 1 – 2 of 2) sorted by relevance
114 struct MatMulShape { struct115 int64 m;116 int64 k;117 int64 n;
110 Status MatMulShape(shape_inference::InferenceContext* c) { in MatMulShape() function