Searched refs:bc_shape (Results 1 – 2 of 2) sorted by relevance
213 TensorShape bc_shape; in broadcast_shape() local215 auto broadcast = [&bc_shape](const TensorShape & other) in broadcast_shape()217 if(bc_shape.num_dimensions() == 0) in broadcast_shape()219 bc_shape = other; in broadcast_shape()225 const size_t dim_min = std::min(bc_shape[d], other[d]); in broadcast_shape()226 const size_t dim_max = std::max(bc_shape[d], other[d]); in broadcast_shape()230 bc_shape = TensorShape{ 0U }; in broadcast_shape()234 bc_shape.set(d, dim_max); in broadcast_shape()241 return bc_shape; in broadcast_shape()
330 TensorShape bc_shape = TensorShape::broadcast_shape(infos.tensor_shape()...); in broadcast_shape_and_valid_region() local331 ValidRegion bc_valid_region{ Coordinates(), bc_shape }; in broadcast_shape_and_valid_region()355 return std::pair<TensorShape, ValidRegion>(bc_shape, bc_valid_region); in broadcast_shape_and_valid_region()