Searched refs:lhs_tf_shape (Results 1 – 1 of 1) sorted by relevance
97 TensorShape lhs_tf_shape; in BroadcastOpsToSame() local99 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(lhs_xla_shape, &lhs_tf_shape)); in BroadcastOpsToSame()101 if (!lhs_tf_shape.IsSameSize(rhs_tf_shape)) { in BroadcastOpsToSame()102 BCast bcast(BCast::FromShape(lhs_tf_shape), BCast::FromShape(rhs_tf_shape)); in BroadcastOpsToSame()