Searched refs:shpy (Results 1 – 3 of 3) sorted by relevance
36 std::vector<int64_t> shpy{3, 4}; in TEST_F() local37 auto y_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shpy); in TEST_F()
263 ShapeVector BroadcastShape(ShapeVector shpx, ShapeVector shpy) { in BroadcastShape() argument264 int dlen = SizeToInt(shpx.size()) - SizeToInt(shpy.size()); in BroadcastShape()271 (void)shpy.insert(shpy.begin(), 1); in BroadcastShape()274 if (shpx.size() != shpy.size()) { in BroadcastShape()280 auto b = shpy[i]; in BroadcastShape()
52 ShapeVector BroadcastShape(ShapeVector shpx, ShapeVector shpy);