Searched refs:shpx (Results 1 – 3 of 3) sorted by relevance
34 std::vector<int64_t> shpx{1, 3}; in TEST_F() local35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shpx); in TEST_F()
263 ShapeVector BroadcastShape(ShapeVector shpx, ShapeVector shpy) { in BroadcastShape() argument264 int dlen = SizeToInt(shpx.size()) - SizeToInt(shpy.size()); in BroadcastShape()267 (void)shpx.insert(shpx.begin(), 1); in BroadcastShape()274 if (shpx.size() != shpy.size()) { in BroadcastShape()278 for (size_t i = 0; i < shpx.size(); i++) { in BroadcastShape()279 auto a = shpx[i]; in BroadcastShape()
52 ShapeVector BroadcastShape(ShapeVector shpx, ShapeVector shpy);