Searched refs:broadcasted_shape (Results 1 – 2 of 2) sorted by relevance
115 broadcasted_shape = shape.broadcast_to_rank(rank)116 self.assertShapeEq(broadcasted_shape, expected)117 self.assertEqual(broadcasted_shape.rank, rank)
528 const Shape& broadcasted_shape = ShapeUtil::MakeShape( in BinaryOp() local532 InDimBroadcast(broadcasted_shape, from, broadcast_dimensions)); in BinaryOp()