Searched refs:lhs_broadcast_shape (Results 1 – 1 of 1) sorted by relevance
89 TensorShape lhs_broadcast_shape(broadcast_helper.output_batch_shape()); in Broadcast() local90 lhs_broadcast_shape.AddDim(m); in Broadcast()91 lhs_broadcast_shape.AddDim(m); in Broadcast()92 auto lhs_output = BroadcastTo(lhs, lhs_broadcast_shape.dim_sizes()); in Broadcast()