Searched refs:rhs_broadcast_shape (Results 1 – 2 of 2) sorted by relevance
98 TensorShape rhs_broadcast_shape(broadcast_helper.output_batch_shape()); in Broadcast() local99 rhs_broadcast_shape.AddDim(m); in Broadcast()100 rhs_broadcast_shape.AddDim(n); in Broadcast()101 auto rhs_output = BroadcastTo(rhs, rhs_broadcast_shape.dim_sizes()); in Broadcast()
983 rhs_broadcast_shape = array_ops.concat([broadcast_batch_shape, [d, m]],987 broadcast_rhs = array_ops.broadcast_to(rhs, rhs_broadcast_shape)991 broadcast_perm = array_ops.broadcast_to(perm, rhs_broadcast_shape[:-1])1001 permuted_rhs = array_ops.reshape(permuted_rhs, rhs_broadcast_shape)