Searched defs:y_shp (Results 1 – 4 of 4) sorted by relevance
31 auto y_shp = y_shape_map[kShape]; in MatMulInferShape() local
38 auto y_shp = y_shape_map[kShape]; in BatchMatmulInferShape() local
278 auto y_shp = y->shape()->shape(); in InferImplMatMul() local344 auto y_shp = y->shape()->shape(); in InferImplBatchMatMul() local
293 std::vector<int64_t> y_shp{768, 768}; in TEST_F() local