Searched refs:BatchMatMulV2Shape (Results 1 – 3 of 3) sorted by relevance
111 Status BatchMatMulV2Shape(shape_inference::InferenceContext* c);
309 Status BatchMatMulV2Shape(shape_inference::InferenceContext* c) { in BatchMatMulV2Shape() function
138 .SetShapeFn(shape_inference::BatchMatMulV2Shape);157 .SetShapeFn(shape_inference::BatchMatMulV2Shape);